前言
之前的系统崩了,换了新的Ubuntu系统,版本号18.04,由于Pytorch 刚刚发布了新的1.1版本,并且支持CUDA 10 ,所以我在这里记录一下安装的过程。虽然是ubuntu18.04和cuda10 ,但安装过程是一样的,理论上这个教程适用所有版本。
需要的文件
- CUDA 10 本地安装的deb 文件
下载地址: 下载地址 - cuDNN 7.5.1的deb 文件
这里需要去NVIDIA的开发者官网注册一个账号,用账号登录之后才能继续下载。注册登录过程我们不表。登陆之后,可以点击下载地址,同意协议之后选择适配CUDA 10 的 cuDNN 7.5.1 下载,选择 cuDNN Runtime Library for Ubuntu18.04 (Deb) 和 cuDNN Developer Library for Ubuntu18.04 (Deb) 两个文件。
注意事项
这里我们不需要再去下载安装 Ubuntu 的Nvidia 驱