赞
踩
在安装pytorch之前,相比你已经安装好了cuda和anaconda,已经配置好了镜像源了,会简单使用conda了。
如果不会,请看我的博客
深度学习环境----CUDA+cudnn(无废话) 安装cuda,cudnn-CSDN博客
深度学习环境------anaconda,anaconda的下载,安装与使用(无废话)-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,按照我画的圈进行勾选,复制代码,
- # CUDA 11.6
- 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
等它安装完就完事,剩下的缺啥安啥
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。