在CUDA10.1版本下安装tensorflow-gpu=1.13.1版本

安装

cuda10.1安装tensorflow-gpu的1版本会出现一些问题,应该在安装之前先输入

conda install cudatoolkit=10.0

在安装cudatoolkit之后再进行安装tensorflow就成功。

测试

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
print(tf.test.is_gpu_available())

print(tf.__version__)

猜你喜欢

转载自blog.csdn.net/qq_43263543/article/details/123360968