1. CPU 环境
1.1 进入官网
选择对应的基本信息,下方即可生成相应的安装命令;
1.2 键入对应命令
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
2. GPU 环境
网址:CUDA Toolkit 10.1 update2 Archive
wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run