Win10+NVIDIA TITAN Xp | GeForce RTX 2060+Python3.6+TensorFlow_gpu2.1.0+keras2.3.1+cuda10.1+cudnn7.6.5.32记录
安装两大步
1、创建虚拟环境+安装GPU+Keras+卸载重装h5py==2.10.0+Anaconda安装spyder
参考TensorFlow环境搭建记录(截止2021-07-13)
2、cuda+cudann
cuda_10.1.243_426.00_win10 + cudnn-10.1-windows10-x64-v7.6.5.32
整体思路参考深度学习环境搭建(GPU)CUDA安装(完全版)
注意
只安装cuda组件,不选其他,cuda安装在D盘,提前建立好路径,不要忘记添加
如果原先有安装一个cuda的话。只需重新选择 安装路径 和 添加 环境变量(相当于一式两份,只要有一个cuda匹配就能够正常运行。参考如何在电脑上安装多个版本的CUDA
一路默认,等待安装完成。
尽量不选择装Visual Studio Integration,不然后面会出现问题
方便复制的东西
D盘的安装位置:
D:\CUDA_10.1\CUDA
D:\CUDA_10.1\CUDA
D:\CUDA_10.1\CUDA Sample
1)环境变量
CUDA_SDK_PATH_V10_1 = D:\CUDA_10.1\CUDA Samples
CUDA_LIB_PATH_V10_1 = %CUDA_PATH_V10_1%\lib\x64
CUDA_BIN_PATH_V10_1 = %CUDA_PATH_V10_1%\bin
CUDA_SDK_BIN_PATH_V10_1 = %CUDA_SDK_PATH_V10_1%\bin\win64
CUDA_SDK_LIB_PATH_V10_1 = %CUDA_SDK_PATH_V10_1%\common\lib\x64
2)系统变量path
%CUDA_LIB_PATH_V10_1%
%CUDA_BIN_PATH_V10_1%
%CUDA_SDK_LIB_PATH_V10_1%
%CUDA_SDK_BIN_PATH_V10_1%
D:\CUDA_10.1\CUDA\lib\x64
D:\CUDA_10.1\CUDA\bin
D:\CUDA_10.1\CUDA Samples\common\lib\x64
D:\CUDA_10.1\CUDA Samples\bin\win64
最后复制cudann到指定文件夹,这里为D:\CUDA_10.1\CUDA
启发:配置GPU环境注意两处对应
参考tensorflow-gpu与cuda、keras之间的对应关系
1)TensorFlow与Keras版本的对应
tf1.15 2.0 2.1 2.2 对应 Keras 2.3.1
2)tensorflow-gpu与cuda之间的对应
tensorflow-gpu 2.1.0 2.2.0 2.3.0 对应 cuda10.1