1.驱动安装
首先先确定是否安装驱动,驱动下载地址,一般情况下,驱动已经有了,这一步可以直接跳过。
2.cuda安装
注意在下载cuda的时候,注意版本,因为pytorch一般不支持最新版本,所以尽量不要下载cuda最新版本,cuda下载地址,将cuda下载之后,需要安装cuda,安装时需要命令行进行的,使用sudo sh 文件名,如图:
接下来基本上选择yes或者接受,除过有一个选项时需要安装驱动的,选拒绝就可以了。
3.cudnn安装
在下载cudnn的时候,需要下载与cuda对应的版本,cudnn下载链接,下载之后,将cudnn中的文件复制到cuda对应的文件夹中(类似于windows中cudnn的方式),只不过ubuntu是需要命令行进行复制的。