Pytorch1.4 GPU 环境

首先确定你的电脑显卡是 NVIDIA 的，如果不是，请安装 CPU 版本。使用 pip 直接安装：

pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

驱动

• 下载 CUDA 驱动CUDA10.1

• 双击安装：
  
  

显卡驱动按照自己情况安装：

根据自己要求，选择安装位置

CUDNN

下载与 CUDA10.1 匹配的CUDNN。需要登录 NVIDIA 开发者社区

• 解压后复制到 CUDA Development 目录

环境变量

查看环境变量是否自动添加了，如果没有就自己添加一下 bin 目录

测试

查看是否安装成功

安装 Pytorch

安装好 anaconda 后，用 pip 工具安装 pytorch：

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

安装好后，测试：

如果返回 True 则 Pytorch GPU 版安装完成。