首页 > 其他 > 详细

3090显卡(CUDA11.1)安装Pytorch

时间:2021-05-03 00:21:01      阅读:75      评论:0      收藏:0      [点我收藏+]

由于官方暂时没有11.1版本的Pytorch安装方法,可以使用11.0版本的安装方法,经过测试,在3090上是可以正常使用的。
官方给出的安装方法:

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

在安装过程中,由于连接的是Pytorch的官方地址,所以下载的速度是很慢的,可以手动下载torch包和torchvison的包进行安装,运行一下上面的安装命令,就能看到地址。
另外,安装完torch包以后,在安装torchvison的包时,可能会自动下载另一个版本的torch包,可以通过

 pip install --no-deps torchvision-0.8.2+cu110-cp38-cp38-linux_x86_64.whl 解决。

3090显卡(CUDA11.1)安装Pytorch

原文:https://www.cnblogs.com/MaggieForest/p/14726716.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!