1、安装pytorch需要提前安装好anaconda
配置anaconda镜像源:
conda config --add channel http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channel http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --add cannel http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64
conda config --set show_channel_urls yes
查看当前使用源:
conda config --show-sources
运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引。
还原原始源:conda config --remove-key channels
2、打开 Anaconda 自带的 Anaconda Prompt:创建虚拟环境
先查看安装的python版本:
conda create --name pytorch python=3.9
注意,这里的 pytorch 是虚拟环境的名称,可随意取。
创建好后会提示:
3、进入pytorch虚拟环境:
activate pytorch
退出虚拟环境:
conda deactivate
4、安装pytorch
4.1 先查看当前计算机的cuda版本(英伟达版本):nvidia-smi
4.2 选择对应的pytorch版本安装
注意:由于cuda版本是11.4在这里就只能选cuda 11.3的版本进行安装
安装命令:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
注意:如果安装过程中报HTTP 000 之类的错误,可以把上面安装命令中的-c去掉。
4.3 安装过程中报错
解决方法:问题出在源的问题上,源文件在C:\Users\zyl\.condarc
把里面的https改成http,并把“- defaults”删除。或者可以直接用下面的源:(复制到.condarc文件里面)
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64
show_channel_urls: true
注意1:如果安装错误了,需要把整个虚拟环境都删除,然后再重新安装:
进入anaconda的控制管理器,选择环境选项--当前虚拟环境pytorch,点击左下角的remove
注意2:在安装pytorch的时候最好不要再点安装界面,这样很容易下载安装包的时候断掉(亲测多次)
5、检验是否安装成功pytorch
能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号
import torch
torch.cuda.is_available()
x=torch.randn(4,4)
print(x)
返回结果如上图,则说明整个过程安装完成。