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