安装PyTorch的详细过程

进入pytorch的官网：点这里

进入官网进行下载

 上面这个情况是需要电脑上有单独的英伟达的显卡、或者英伟达的显卡和集显这两种情况都是可以的。先查看一下自己的，如果可以直接复制上面的那串代码。

 

 

打开命令执行框输入

nvidia-smi

查看当前驱动的版本号，观察Driver Version的值是否大于400，如果小于请更新显卡驱动。 

接下来就是安装pytorch

打开这个右键以管理员身份运行，切换到之前下载的pytorch的虚拟环境中。 

输入下面这个代码，将环境切换到pytorch的虚拟环境中

activate pytorch

 然后把刚才复制的代码，输入进去

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 安装完成

 

检验是否安装成功

 输入下面这个代码，将pip全部列出来，找到torch

pip list

 然后进入python，调用torch库

输入下面的代码

import torch

可以看到什么也没有显示出来，然后再输入一行代码

torch.cuda.is_available()

当返回ture就表示安装完成。