win10+cuda10.1+cudnn7.5+anaconda 3 tensorflow-gpuのインストール
2889 ワード
import tensorflow as tf
このpip命令は最新バージョンのtensorflow 1に従う.4(2のbetaバージョンを除く).よし、やっと問題が来た...ImportError: DLL load failed: 。
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/errors
for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.
ネットGitHubで調べてみると、大勢の人がこの問題を起こした.Win 10:ImportError:DLL load failed:一度見てから問題を発見した.「confirmed it needs a dependency on CUBLAXX_100.DLL
.Then I installed CUDA 10.0 and it works.」cudaバージョンが違います...まさかcudaを再ロードするの?cudaを10.0に降格しますか?今はcudaを組み立てるのが簡単で、追加の配置も必要ありませんが、やはり面倒ですね.前はよく使っていました.ちょっと悔しいですね.import tensorflow as tf
import numpy as np
b = tf.test.is_gpu_available(cuda_only = False, min_cuda_compute_capability = None)
print(b)
# NumPy (phony data), 100 .
x_data = np.float32(np.random.rand(2, 100)) #
y_data = np.dot([0.100, 0.200], x_data) + 0.300
#
#
b = tf.Variable(tf.zeros([1]))
W = tf.Variable(tf.random_uniform([1, 2], -1.0, 1.0))
y = tf.matmul(W, x_data) + b
#
loss = tf.reduce_mean(tf.square(y - y_data))
optimizer = tf.train.GradientDescentOptimizer(0.5)
train = optimizer.minimize(loss)
#
init = tf.global_variables_initializer()
# (graph)
sess = tf.Session()
sess.run(init)
#
for step in range(0, 201):
sess.run(train)
if step % 20 == 0:
print( step, sess.run(W), sess.run(b))
成功出力結果!!大功を成し遂げる.- Windows 10 Home 64bit
- CUDA Toolkit 10.0 (Sept 2018)
- Download cuDNN v7.6.0 (May 20, 2019), for CUDA 10.0
- Python 3.7.3 Windows AMD64
- Tensorflow-gpu 1.13 1.14 2
---------------END-------------------------
2019年7月23日