python和troch有啥区别_python

下载完之后,直接一直next安装,然后得到的结果如图所示:


python和troch有啥区别_cuda_02

然后,我们按照视频中的,创建pytorch环境:

conda create -n pytorch python=3.6

成功创建pytorch环境后,可以输入

activate pytorch

来进入环境,会发现命令行前面的base变成了pytorch


python和troch有啥区别_python_03

然后输入pip list,可以看到现在的package的列表:


python和troch有啥区别_cuda_04

可以看到,现在还没有pytorch,接下来我们将进入pytorch的安装。

显卡驱动的安装

在安装pytorch之前,我们需要安装显卡的驱动

(这个驱动我刚装完电脑就已经装好了,所以下面只是用截图简单示意一下,并不是实时的演示。)

(说好了记流水账的。。。这一节稍微跑偏了):

进入Nvidia的官网


python和troch有啥区别_python_05

搜索到了显卡驱动以后,直接下载,并且一直“下一步”无脑安装即可。安装完之后,可以看到任务管理器-性能里面出现了GPU的一栏:


python和troch有啥区别_python_06

说明这个时候驱动已经安装完毕。(继续流水账)

Pytorch的安装

首先,我们到达pytorch的官网https://pytorch.org/,选择想要的pytorch版本


python和troch有啥区别_python和troch有啥区别_07

我们的系统是windows,package当然选conda,语言当然是python,由于之前我已经安装了11.1的CUDA,所以这里的CUDA版本选择CUDA 11.1。我们可以看到下面的命令为:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

其中,前面的"pytorch torchvision torchaudio cudatoolkit=11.1"都是软件名称,后面的-c pytorch -c conda-forge意识是下载的源。因此,这里是直接从pytorch的源或者conda forge进行下载。虽然很慢,于是我在运行上一个命令之前,加载了清华的镜像源:

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

然后紧接着,我用了下面的指令来安装:

conda install pytorch torchvision torchaudio cudatoolkit=11.1

这个时候,我以为会从清华镜像源来下载pytorch,没想到清华镜像源里面似乎没有这个版本的cudatoolkit:


python和troch有啥区别_cuda_08

所以,我干脆老老实实地输入下面的命令,并且死等:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

python和troch有啥区别_cuda_09

这个下载速度着实慢的一批,我甚至还去踢了一场球,回来再一看,发现pytorch直接没下完,安装失败。

查阅资料,我发现我没有添加conda-forge的清华镜像源,导致下载失败,于是我输入:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

然后这个时候再进行安装:

conda install pytorch torchvision torchaudio cudatoolkit=11.1

终于成功了!


python和troch有啥区别_cuda_10

输入python,再import torch来验证安装是否成功


python和troch有啥区别_python_11

没有提示就是最好的提示!

此时我们pip list,来看一下现在的package列表:


python和troch有啥区别_python和troch有啥区别_12