【从零开始复现PVnet】(上)搭建项目系统环境(CUDA+cuDNN+anaconda+pyCharm)

本文详细介绍了如何在Ubuntu18.04系统上为GTX1060显卡安装nvidia-driver-470,搭配CUDA10.2、cudnn,并配置PyCharm开发环境,以便于实现如PVNet位姿估计等深度学习项目的复现。

环境:
ubuntu18.04
GPU:GTX1060
显卡驱动版本:nvidia- driver- 470
pytorch 1.5.1 + torchvision0.6.0+ cuda10.2

参考链接:

ubuntu安装cuda-10.2以及对应版本的cudnn

PVNet位姿估计网络简单复现2022-5-21

Ubuntu18.04系统搭建深度学习环境

【复现笔记】clean-pvnet复现

1.安装显卡驱动和CUDA

1.1安装显卡驱动

先更新源

sudo apt update

安装ubuntu-drivers,我们可以通过ubuntu-drivers检测你的NVIDIA显卡型号和推荐的驱动程序的模型。

sudo apt install ubuntu-drivers-common

获得NVIDIA显卡型号和推荐的驱动程序

ubuntu-drivers devices

我的系统推荐的显卡驱动版本 nvidia-driver-470,使用下面代码安装

sudo apt install nvidia-driver-470

Nvidia驱动默认安装OpenGL桌面,然而这又与GNOME桌面冲突,为了系统不宕机,需禁用nvidia的OpenGL。Linux系统一般默认安装的是开源的nouvea显卡驱动,它与nvidia显卡驱动产生冲突,欲装nvidia必禁nouvea!
禁用nouveau驱动:

sudo gedit /etc/modprobe.d/blacklist.conf

在弹窗最后加入下面代码:

sudo gedit /etc/modprobe.d/blacklist.conf

保存,关闭弹窗;至此,安装完毕,重启

sudo reboot

打开终端检查是否禁用成功,执行下面命令若无输出代表禁用nouveau生效

lsmod | grep nouveau

测试安装是否成功,这会返回你的所有显卡的信息

nvidia-smi
1.2下载完整版cuda

我选择CUDA10.2版本,进入nvidia的CUDA驱动下载页:
https://developer.nvidia.com/cuda-downloads

在这里插入图片描述
在这里插入图片描述

通过指令下载

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Fate-Sky

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值