在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_avail
原创
2022-03-08 19:30:26 ·
2542 阅读 ·
0 评论