本方法不需要去找对应的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())

成功显示如下:
在这里插入图片描述
成功!

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐