Ubuntu 16.04环境

1.安装nvidia 驱动+CUDA+cuDNN

对应版本安装
二. pytorch环境安装
测试CUDA nvidia-smi出现显卡信息,则安装成功。

2.安装anaconda3

二. pytorch环境安装
下载anaconda的sh文件,然后安装 sudo bash ***.sh

3.安装pytorch

二. pytorch环境安装
一般使用conda intall 或者pip intall
注意匹配CUDA版本,旧版本点击选择 Previous versions of PyTorch

相关文章: