WIN10环境下深度学习中使用torch时无法启用GPU的一些坑

使用DeepCTR-zorch来进行深度学习时,

torch.cuda.is_available()

返回值总是false而导致无法使用GPU来进行计算。这对笔者这拥有号称卡皇1080Ti的人来说,简直太不爽了。于是网上找了找资料,发现都介绍的不全面,所以在看本文的时候,当你尝试了网上所说的驱动版本问题,CUDA版本问题,环境变量设置问题以及各种莫名其妙的版本问题后再来看本文,本文假定您基本已安装了其他所有该依赖的相关文件。

先说本文所使用的的环境

Python 3.9

CUDA V11.3  

下载链接:https://developer.nvidia.com/cuda-toolkit-archive

cuDNN

下载链接:https://developer.nvidia.com/rdp/cudnn-download

下载完后注意依据文档,直接将解压后的三个目录,直接拷贝覆盖到CUDA安装路径下的同名目录即可。

===========================分割线===========================

Anaconda3-2021.05-Windows-x86_64

下载地址:https://www.anaconda.com/distribution/

Miniconda3-py39_4.9.2-Windows-x86_64

下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

 

最后,使用conda安装tensorflow的GPU版本,注意不要使用PIP,可能引起版本冲突。

conda install tensorflow-gpu

OK,现在返回值是True了,再次跑一次程序,终于,GPU工作了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值