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则表示成功

原文地址:https://www.cnblogs.com/chenlove/p/14119334.html