栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

安装pytorch-gpu版

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

安装pytorch-gpu版

目录

首先,安装cuda

然后安装pytorch


之前在清华源下载的pytorch是cpu版的 在python下测试torch.cuda.is_available()返回的是false

故在万能的Google下,找到了相关文章,进行整理

首先,安装cuda

没有英伟达控制面板 建议下载一个

然后到官网去下载,我的cuda版本是11.4 ,目测可以下载11版本的,目前暂不知道version那一栏后面的server的意思(但因为看不懂这上面version的意思,我按照我看的pytorch视频来安装了version为10的版本,目前暂未发现问题)

 也可以去CUDA Toolkit 11.4 Downloads | NVIDIA Developer这里下载对应版本

在网上看到还要安装cuDNN,但是我没安装,后续出现问题,再补回来(也可能安装了我不知道,萌新脸)

参考博客:Win10 下安装 CUDA Toolkit ( for PyTorch )

参考博客:win10安装CUDA和cuDNN的正确姿势 - 知乎

然后安装pytorch

注意,这里不要用清华源的下载连接

如果清华源可以的话,还是建议通过清华源下载,毕竟速度快

附清华源下载教程

否则你看conda-list或者pip list时,会发现cpu only或带有cpu字样的数据

建议本地安装,在线安装可能会有网络错误等等

去官网本地安装包进行下载,如果进不去,那就是被那啥了,需要那啥可以私信我(可能不及时回)

然后我听网上说下载速度慢,但是我下载速度挺快的,估计是代理的缘故,所以慢慢等待下载好吧。

一开始进去那个本地安装包网址,可能发现里面的版本之,不知道下载什么,我是通过先

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

然后他给我报错我发现我要的安装包是这个

 (如果他没报错,你就慢慢等他下载完就行,可能是我网络波动,或者你多试试几次)

然后就利用ctrl+F寻找

 把名字对上就行,不知道的话就去官网看版本,然后下载对应版本的软件,目前不知道后缀cp36/cp37/cp38有什么区别,所以只能一个个试了(所以能用pip在线安装就在线安装吧)

 等他下完后,在你的虚拟环境中输入 pip install 你的下载文件的地址

 然后弄完三个后pytorch、torchvision、torchaudio即可。

 

torch.cuda.is_available() 

大功告成!

参考博客:torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-程序员宅基地 - 程序员宅基地

参考博客:解决torch.cuda.is_available()一直返回False的玄学方法之一_dysljxjsq的博客-CSDN博客

参考博客:

Anaconda下pip离线安装Pytorch(亲测可用)_caracol的博客-CSDN博客_离线安装pytorch

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/341745.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号