RTX4060安装pytorch-gpu (Cuda12.2),[pytorch] torch.cuda.is_available() False,torch使用GPU运行,而不是CPU运行的操作指南

本文介绍了如何在拥有RTX4060Ti显卡和CUDA12.2环境下安装和配置PyTorch,包括CUDNN版本选择、CUDA工具验证、环境设置(如Conda环境创建与管理),以及性能提升带来的测试对比。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

RTX4060安装pytorch-gpu (Cuda12.2)技术分享

问题描述:
在cmd中运行以下命令返回的是false

在这里插入图片描述
第一步:我的电脑显卡是4060Ti,我安装的cuda是12.2,那么我需要下载的cudnn的版本是8.9.3
在这里插入图片描述
下载的具体网址
在这里插入图片描述
然后下载的cudnn,将三个文件夹中的文件分别复制粘贴到cuda安装的对应的目录下
在这里插入图片描述
在这里插入图片描述cmd中查看是否安装成功

nvidia-smi

在这里插入图片描述

nvcc -V

在这里插入图片描述

确认一下是否安装成功
进入如下目录:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\demo_suite

在这里插入图片描述

执行 bandwidthTest.exe,看见PASS
在这里插入图片描述

执行 deviceQuery.exe,看见PASS
在这里插入图片描述
操作就是先打开命令行cmd,然后将以上的两个.exe拖进cmd中,回车就能看到结果了

如果以上都没有问题,那就需要设置一下镜像
设置镜像
在C盘用户目录下,加入如下内容即可。

在这里插入图片描述

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple/
[install]
use-mirrors =true
mirrors =http://pypi.mirrors.ustc.edu.cn/simple/
trusted-host =pypi.mirrors.ustc.edu.cn

创建pytorch环境,我当前的是3.8.8,看第一幅图有版本。

在cmd中

conda create –n pytorch38 python=3.8
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

安装的过程有点慢,耐心等待即可。

完成安装时,在cmd中输入pip list时,查看安装的版本
在这里插入图片描述
安装的路径在下图:
在这里插入图片描述
以前安装的是torch的版本是cpu版本,我先剪切出去了,以防不成功
在这里插入图片描述
cmd中,进入python环境后,输入一下的测试命令,进行测试,结果:
在这里插入图片描述
conda 的基础命令

#查看现有环境
conda env list
#更新conda
conda upgrade --all
#使用py36环境
conda activate py36
#返回base环境
conda deactivate
#创建一个名字叫py36的环境,版本为3.6
conda create -n py36 python=3.6
#删除一个名字叫py36的环境
conda remove -n py36 --all

进行项目上的测试结果返回:
在这里插入图片描述
100Epochs只需要1分钟,我用cpu跑需要30分钟。差了30倍的速度

最新的执行的命名:

pip install torch==2.4.1+cu128 torchvision==0.19.1+cu128 torchaudio==2.1.1+cu128 --index-url https://download.pytorch.org/whl/cu128

安装命令:

pip install torch==2.4.1+cu124 torchvision==0.19.1+cu124 --index-url https://download.pytorch.org/whl/cu124

pip install torchaudio==2.4.0+cu124 --index-url https://download.pytorch.org/whl/cu124

评论 39
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

骨子里的偏爱

上传不易,还请多多支持。

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值