conda 安装GPU——CUDA

linux查看cuda版本

cuda一般安装在 /usr/local/cuda/ 路径下,该路径下有一个version.txt文档,里面记录了cuda的版本信息

cat  /usr/local/cuda/version.txt

然后安装对应的cuda

conda install cudatoolkit=10.0

 

最后测试是否安装成功

import tensorflow as tf
tf.test.is_gpu_available()
Out[1]: True

显示True则表示成功

posted @ 2020-12-11 11:12  Python研究者  阅读(3436)  评论(0编辑  收藏  举报