当前位置:   article > 正文

深度学习环境-------pytorch安装------超详细_pytorch1.13.1支持cuda版本

pytorch1.13.1支持cuda版本

安装pytorch之前,相比你已经安装好了cuda和anaconda,已经配置好了镜像源了,会简单使用conda了。

如果不会,请看我的博客

深度学习环境----CUDA+cudnn(无废话) 安装cuda,cudnn-CSDN博客

深度学习环境------anaconda,anaconda的下载,安装与使用(无废话)-CSDN博客

 conda加镜像源,conda配置镜像源-CSDN博客

如果以上都做好了,就开始安装torch

搜索pytorch  进入官网          PyTorch

 在这之前,先看一下自己的cuda版本

我建议直接在电脑里查就可以

打开c盘   ,打开Program Files文件夹

打开NVIDIA GPU Computing Toolkit文件夹

打开cuda,里面显示的是几,你就安装了哪个版本

查找自己的版本

我的是11.6版本,以我的版本为列去安装pytorch

打开pytorch官网后,往下翻,找到这个    Previous versions of PyTorch

然后就往后翻就可以了,找cuda11.6

我比较喜欢pip安装,pip加个镜像源,下载速度很快,以pip安装为例

pip镜像源安装指导  pip配置镜像源-CSDN博客 )

 

就目前我的cuda11.6的来说最新的能安装的pytorch版本为1.13.1,按照我画的圈进行勾选,复制代码,

  1. # CUDA 11.6
  2. pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

在conda环境里直接站上去就可以了

 安装指导

如果自己还没有创环境的话,还不会用conda,看这个

深度学习环境------anaconda,anaconda的下载,安装与使用(无废话)-CSDN博客

 比如我创的环境名字为whh,python版本为3.8的

conda create -n whh python=3.8

安装好了之后,把代码复制进去

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

等它安装完就完事,剩下的缺啥安啥

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/193906
推荐阅读
相关标签
  

闽ICP备14008679号