300字范文,内容丰富有趣,生活中的好帮手!
300字范文 > Ubuntu 搭建 NVIDIA驱动 CUDA 和 pytorch GPU 环境

Ubuntu 搭建 NVIDIA驱动 CUDA 和 pytorch GPU 环境

时间:2022-01-24 15:56:14

相关推荐

Ubuntu 搭建 NVIDIA驱动 CUDA 和 pytorch GPU 环境

首先,查看自己的驱动型号

输入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

安装成功

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。