1. 前期准备工作
下载 Anaconda3,PyCharm,NVIDIA GeForce GTX 1080Ti 显卡驱动,CUDA,cuDNN,Pytorch 安装包,保存到同一个目录(如“下载”目录)
1.1 Anaconda3:
https://www.anaconda.com/distribution/
选择 Linux Python 3.7 version 64-Bit (x86) Installer,如下图:
1.2 PyCharm:
https://www.jetbrains.com/pycharm/download/#section=linux
选择 Community 版,如下图:
1.3 NVIDIA GeForce GTX 1080Ti 显卡驱动
https://www.geforce.cn/drivers
按以下条件筛选:
一般选列表第一个:
点击下载按钮:
1.4 CUDA
1.5 cuDNN
https://developer.nvidia.com/rdp/cuDNN-archive
下载这个要向注册一个NVIDIA 开发者账号,微信扫码就行。
按以下条件筛选并下载:
1.6 PyTorch
两个 whl 包:
https://download.pytorch.org/whl/cu92/torch-1.3.0%2Bcu92-cp37-cp37m-linux_x86_64.whl
https://download.pytorch.org/whl/cu92/torchvision-0.4.1%2Bcu92-cp37-cp37m-linux_x86_64.whl
如果要安装其他版本,自行到 https://download.pytorch.org/whl/torch_stable.html 下载。