下载前记得查看tensorflow-gpu与CUDA、cuDNN对应的版本
https://tensorflow.google.cn/install/source_windows?hl=en#gpu
tensorflow-gpu2.10.0
pip install tensorflow-gpu
后面也可以跟上对应的版本号,如果下载失败可能是网不好,多试几次,也可以找找其他教程。
CUDA11.2
下载:https://developer.nvidia.com/cuda-toolkit-archive
一路下一步
a few moments later
安装成功后,我的电脑上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了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