一、 用pytorch官网安装的详细教程见:
https://wwwblogs/zhouzhiyao/p/11784055.html
这篇文章讲的很详细,但是这样安装的话速度非常慢,经常会失败,下面我们讲使用清华镜像来快速安装pytorch。
二、在https://wwwblogs/zhouzhiyao/p/11784055.html
所给出的教程中安装到第5步《5. 安装 PyTorch 准备》的时候不要去打开pytorch官网。然后,按照以下步骤进行:
1、在Anaconda Prompt中依次输入以下命令
conda config --add channels http://mirrors.tuna.tsinghua.edu/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels http://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/
必须注意:(1)确保左侧为pytorch
(2)网址中一定是http,而不是https,很多教程上是https,执行的过程中会报错,这里一定一定要注意。
2、然后再打开pytorch官网https://pytorch/,选择适合自己电脑的版本,PyTorch Build 选择 Stable;选择系统;Package,Windows下推荐 conda,Linux 下推荐 pip;选择自己电脑支持的CUDA,我的CUDA 选择10.2。如果没有显卡的话,选择 None。我的选择如图所示:
复制下面Run this Commad中的代码:conda install pytorch torchvision torchaudio cudatoolkit=10.2,在Anaconda Prompt中运行即可。
*****注意:***一定要将官网上conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch 命令中的“-c pytorch ”去掉,要不然还是默认在官网上下载安装,不会在镜像中下载,可能会导致失败!
3、验证是否成功
(1)在命令行左边为 pytorch 环境中,输入 python
(2)之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了
(3)接下来,输入 torch.cuda.is_available(),如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。
(4)输入exit()退出python交互环境
4、如果不成功可以再重复一遍
但是在重试之前先要把已经加载的链接删除,要不会出现重复加载的错误,出现如下提示:
具体做法:
1、找到用户下一个叫.condarc的文件。文件路径在:C:\Users\Administrator(这是我的路径)下,用记事本打开.condarc的文件,
2、将.condarc文件中的默认channels删掉,
删掉后如下图所示,然后保存,记得一定要保存。
3、再重新从本页中的第1步重新执行一遍,不会报错了!
更多推荐
win10环境下用清华镜像快速安装pytorch
发布评论