ubuntu配置pytorch以及cuda

本文介绍了如何在Ubuntu系统上管理和安装CUDA11.3,同时处理了与PyTorch的兼容性问题,重点强调了在安装NVIDIA驱动前需禁用系统自带的nouveau显卡驱动。
摘要由CSDN通过智能技术生成

大家好啊,今天给大家带来一些想看的东西。

因为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的驱动程序了

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值