首先,查看自己的驱动型号
输入lspci,如上图所示,我的是Quadro K4200
搜索适合的驱动,并下载
记住存放的路径
接下来安装驱动:
参考:/kellyroslyn/article/details/91448848
1)删除、禁用驱动
删除旧的
sudo apt-get purge nvidia*
禁止自带的nouveau nvidia驱动
在/etc/modprobe.d/目录中找到黑名单
添加禁用 信息
blacklist nouveau options nouveau modeset=0
保存退出
更新
重启电脑
重启之后独显没驱动了,界面可能变得很丑,但是不影响使用
进入文本输入模式,按ALT+CTRL+F1
输入Ubuntu的账号,密码
然后,设置权限,找到run存在的位置,并运行驱动安装程序
会自动进入安装,界面操作
安装结束输入输入
sudo nvidia-smi
二、接下来安装anaconda,这一步很简单
参考:/kellyroslyn/article/details/91402195
三、安装CUDA
登录官网:/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal
下载到指定文件夹后,运行
一下几条命令
第二步的时候需要去路径中找自己的版本
按顺序操作,运行结束
三、安装pytorch
下滑选择对应版本,python的版本可以再conda环境下输入python查看
下载太慢的话,参考:/qq_40971025/article/details/101448368
安装成功
输入python
安装成功