Win10+NVIDIA TITAN Xp | 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
只选择“CUDA”
更换安装路径

一路默认,等待安装完成。
尽量不选择装Visual Studio Integration,不然后面会出现问题
尽量不选择装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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值