最近在win10装torch，发现没有Linux那么便捷，用conda和pip装默认版本都是cpu，而且下载速率很慢。网上的办理方法有好几种，但是初学者大概面对的题目是：我的系统该装哪个版本的？cuda、torch、torchvision以及 torchaudio装哪个版本？以及装了cpu版本的显示torch.cuda.is_available() False 该怎么办？ 那么，方法如下（亲测乐成好用） 第一步：创建情况 （-n +情况名（torch) + python版本（3.10）） [code]conda create -n torch python=3.8[/code]第二步：了解你电脑的cuda 版本，在terminal中输入：[code]nvidia-smi[/code] 即可，如下图所示，版本为12.3
第三步：去pytorch官网看你的conda适配的torch、torchvision以及 torchaudio版本，这个链接： Previous PyTorch Versions | PyTorch 比你的CUDA低版本的都能安装，看好对应的版本就可以了，不消官网的命令安装（会很慢）
第四步：开始安装，鉴于使用conda的各种题目，我们这里不使用conda安装，使用pip安装，根据刚才看好的版本，在创建的torch情况下输入命令： [code]#进入torch情况 conda activate torch[/code] [code]#用镜像源快速安装对应版本 pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple [/code]第五步：验证torch安装是否乐成，import torch乐成阐明安装乐成，假如torch.cuda.is_available()返回False 时，阐明装了cpu版本的。不要慌，往下看。
第五步：查看torch情况下所有库： [code]conda list[/code]可以看到，pytorch是cpu版本的
第六步：手动去清华源下载对应的gpu版pytorch，和上面图上看到的cpu的版本保持同等即可 Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror 好比pytorch版本是1.10.1（必要对应），情况创建时python版本是3.8 （必要对应），cuda 12.3（比它小的都可以），那么下面几个可以下载，选一个下载，最好装到一个英文路径下。
第七步：离线安装刚刚下载的文件 [code]# cd 到下载的文件目录，例如安装目录是E:\Download cd E:\Download[/code] [code]#离线安装 conda install --offline pytorch-1.10.0-py3.6_cuda10.2_cudnn7.6.5_0.tar.bz2[/code]末了一步： conda list, 查看pytorch 是否已经被更换。可以看到更换成GPU版本了
再输入 torch.cuda.is_available() 验证，返回True .大功告成！ 总结 到此这篇关于快速安装gpu版本的pytorch的文章就介绍到这了,更多相干快速安装gpu版本pytorch内容请搜索脚本之家从前的文章或继续欣赏下面的相干文章希望各人以后多多支持脚本之家！ 来源：https://www.jb51.net/python/328044wn8.htm 免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！ |
/6 
|手机版|小黑屋|梦想之都-俊月星空
( 粤ICP备18056059号 )|网站地图
GMT+8, 2025-7-1 19:00 , Processed in 0.032054 second(s), 19 queries .
Powered by Mxzdjyxk! X3.5
© 2001-2025 Discuz! Team.
