本方法不需要去找对应的cuda和cudnn版本,也不需要去下载这两个文件,直接从python库中下载对应版本将其直接下载到环境中,直接上代码:
打开Anaconda Prompt
1.创建新环境
conda create -n tensorflow-gpu python==1.13.1
2.切换到新环境
conda activate tensorflow-gpu
3.安装cuda(这里其实要注意自己的cuda版本是否大于这个10.0,可以去cmd nvidia-smi去查看如下图我这是11.7)
conda install cudatoolkit=10.0
4.安装tensorflow-gpu
conda install tensorflow-gpu=1.13.1
5.测试
import tensorflow as tf
print(tf.test.is_gpu_available())
成功显示如下:
成功!