Pytorch 的安装

GPU版本的安装

• Windows平台 CPU 版本安装

conda install pytorch torchvision cpuonly -c puython

　　Windows平台需安装VC，需要的联系博主。

• Linux平台 CPU 版本安装

conda install pytorch torchvision cpuonly -c pytorch

• MacOS平台 CPU 版本安装

conda install pytorch torchvision -c pytorch

GPU版本的安装

• GPU版本需有 NVIDIA 显卡硬件支持，请确保您的显卡支持 cuda，具体型号可查询如下网址：https://developer.nvidia.com/cuda-gpus
• 检查显卡驱动

　　在命令行或者shell中执行：

 nvidia-smi     

　　执行完会打印出您的显卡信息：

　　　　　　

　　在右上角，显示您的显卡目前的cuda驱动版本，比如上图显示为11.4，要注意，这里并不是表明您已经安装了cudatookit，仅仅代表，您当前驱动可以支持的 cuda 最高版本。

• 通过pytorch官网选择安装命令，打开pytorch官网（https://pytorch.org/），在网页中间位置，有安装提示栏，如下所示：

　　　　　　

　　这里本人电脑采用如下命令安装：

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

• 检验Pytorch安装完成：

　　命令行进入 python 使用：

>>> import torch
>>> torch.randn(4,5)

　　若生成结果，安装完成。

• 验证GPU安装

>>> torch.cuda.is_available()

　　返回 True 则安装完成。