纯小白安装pytorch(快速上手)

Scan sprink2024-09-17 20:36

1.首先进入你的虚拟环境(不进入也没关系)

在anaconda prompt中打开，输入activate 虚拟环境

2.查看自己的conda源

conda config --show channels

3.清空自己的conda源

conda config --remove-key channels

4.添加源

bash 复制代码

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --set show_channel_urls yes

5.去Pytorch官网下载对应的cuda版本的pytorch

bash 复制代码

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3

我的cuda版本是11.6(使用nvidia-smi)，我选择比11.6版本低一些的带cuda版本的pytorch

6.如果遇到如下问题的解决办法

问题一

bash 复制代码

 CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://repo.anaconda.com/pkgs/main/win-64/mkl-2023.1.0-h6b88ed4_463 58.conda> Elapsed: -  An HTTP error occurred when trying to retrieve this URL. HTTP errors are often intermittent, and a simple retry will get you on your way.

解决办法

首先检查自己的网是否是通的

bash 复制代码

ping www.baidu.com

通的话就不用管，不通就看一下如何修复(本人没有遇到此问题)

检查防火墙，然后关闭防火墙

bash 复制代码

# 在管理员的身份运行cmd窗口
netsh advfirewall set allprofiles state off

如果关闭防火墙还不行，则把ssl证书给关闭

1.首先在anaconda prompt命令窗口中查看自己的ssl证书

bash 复制代码

conda config --show|grep ssl*

2.然后关闭ssl

bash 复制代码

conda config --set ssl_verify no

3.想开启的话命令就是

bash 复制代码

conda config --set ssl_verify yes

以上就应该能解决报错的问题了