Linux安装cuda环境

发布时间 2023-12-12 17:19:24作者: klaycsu

安装cuda驱动时

需要先卸载开源的cuda驱动
参看是否安装了开源驱动nouveau

lsmod | grep nouveau

关闭

cat <<EOF | sudo tee /etc/modprobe.d/blacklist-nouveau.conf
blacklist nouveau
options nouveau modeset=0
EOF

重新生成内核initramfs

sudo update-initramfs -u

重启

sudo reboot

如果容器出错找不到GPU
Alt text
docker logs contrain 出现
Failed to start Load Kernel Modules
解决方法:
Installing the NVIDIA Container Toolkit

再重启docker服务