大家好啊,今天给大家带来一些想看的东西。
因为cuda以及cudnn与pytorch是配套的,所以想好自己要装哪个版本再开始哈。
本人以前使用的是11.3,此处就装11.3了
激活环境
运行上面的即可安装
cuda的选项
注意!ubuntu系统安装一般会自带驱动,在安装NVIDIA驱动以前需要禁止系统自带显卡驱动nouveau:可以先通过指令lsmod | grep nouveau查看nouveau驱动的启用情况,如果有输出表示nouveau驱动正在工作,如果没有内容输出则表示已经禁用了nouveau。
打开终端,在终端输入下面指令
sudo gedit /etc/modprobe.d/blacklist.conf
之后弹出blacklist.conf文件:
在blacklist.conf文件末尾加上这两行,并保存:
blacklist nouveau
options nouveau modeset=0
然后在终端输入
sudo update-initramfs -u #应用更改
重启,就禁止了ubuntu20.04自带的nouveau显卡驱动了,接下来我们就可以安心的安装自己想要的NVIDIA的驱动程序了