【环境】NVIDIA驱动安装+cuda11.0(ubuntu16.04)
目录
一、N卡驱动的安装
二、CUDA安装
一、N卡驱动的安装
1、去NVIDIA官网找到对应自己显卡驱动的文件
example:GeForce GTX 1060 6G
2、根据pytorch对应版本选择显卡驱动版本,一般为向下兼容,下载run文件,如NVIDIA-Linux-x86_64-460.67.run
3、禁用自带的驱动
打开配置文件:
$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
禁用nouveau,在...
原创
2021-08-31 11:00:29 ·
2502 阅读 ·
0 评论