Ubuntu 16.04环境 1.安装nvidia 驱动+CUDA+cuDNN 对应版本安装 测试CUDA nvidia-smi出现显卡信息,则安装成功。 2.安装anaconda3 下载anaconda的sh文件,然后安装 sudo bash ***.sh 3.安装pytorch 一般使用conda intall 或者pip intall 注意匹配CUDA版本,旧版本点击选择 Previous versions of PyTorch 相关文章: