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

本文详细介绍了在Ubuntu系统中如何安装CUDA和CUDNN,以及后续的PyTorch环境配置。包括驱动检查、CUDA与CUDNN的下载与匹配、显卡信息查看、PyCharm与Anaconda的安装、虚拟环境创建和PyTorch的版本选择安装步骤。
最低0.47元/天 解锁文章
2858

被折叠的 条评论
为什么被折叠?



