1.检查是否有合适的GPU, 如有安装Cuda与CuDNNhtml
2.下载pytorch安装文件,进行pytorch的安装命令行
在桌面上右击若是能找到NVIDA控制面板,则说明该电脑有GPU。控制面板以下,并经过查看系统信息获取支持的Cuda版本htm
点击 帮助->点击 系统信息 弹出下面的对话框,在驱动程序版本那一栏就能看到该计算机使用的驱动版本。blog
官网:https://developer.nvidia.com/cuda-10.1-download-archive-update2get
在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里能够查询到咱们应该下载哪一个版本it
而后,根据实际状况选择合适的版本io
官网 https://developer.nvidia.com/rdp/cudnn-downloadtable
1.与安装其余的软件相似class
2.安装结束后将 ~/nvcc/bin(由于版本的不一样可能在不一样的地方) 目录添加到环境变量变量
3.在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功
4.将CuDNN压缩包解压后,下面的三个文件夹复制到Cuda的安装目录下
5.而后执行Demo, 若是Demo中显示PASS则说明安装成功
官网 https://download.pytorch.org/whl/torch_stable.html 选择合适的版本
torch/torchvision 都须要安装
经过上面输出的信息,说明安装是成功的