tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2

下载前记得查看tensorflow-gpu与CUDA、cuDNN对应的版本

https://tensorflow.google.cn/install/source_windows?hl=en#gpu
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第1张图片

tensorflow-gpu2.10.0

pip install tensorflow-gpu
后面也可以跟上对应的版本号,如果下载失败可能是网不好,多试几次,也可以找找其他教程。

CUDA11.2

下载:https://developer.nvidia.com/cuda-toolkit-archive
一路下一步

tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第2张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第3张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第4张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第5张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第6张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第7张图片a few moments later

tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第8张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第9张图片
tensorflow2.10.0+CUDA11.2+cuDNN8.1 for cuda11.2_第10张图片

安装成功后,我的电脑上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V11_2两个环境变量。
在这里插入图片描述还需要添加如下几个变量:

在用户变量Path下添加:C:\ProgramData\NVIDIA GPU Computing Toolkit\v11.2

在系统变量中添加如下几个变量:

 CUDA_SDK_PATH = C:\ProgramData\NVIDIACorporation\CUDA Samples\v9.2

 CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 

CUDA_BIN_PATH = %CUDA_PATH%\bin 

CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 

CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

cuDNN8.1 for cuda11.2

下载(需要先注册):https://developer.nvidia.com/rdp/cudnn-download
选择对应的版本下载,解压cudnn-11.2-windows10-x64-v8.1.0.77.zip,将文件夹里的内容拷贝到CUDA的安装目录并覆盖相应的文件夹,CUDA默认安装目录:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

参考

1:https://blog.csdn.net/wwtor/article/details/80603296

你可能感兴趣的:(#,工具安装,tensorflow,深度学习,python)