1. 前期准备工作

下载 Anaconda3，PyCharm，NVIDIA GeForce GTX 1080Ti 显卡驱动，CUDA，cuDNN，Pytorch 安装包，保存到同一个目录（如“下载”目录）

1.1 Anaconda3：

https://www.anaconda.com/distribution/
选择 Linux Python 3.7 version 64-Bit (x86) Installer，如下图：

1.2 PyCharm：

https://www.jetbrains.com/pycharm/download/#section=linux
选择 Community 版，如下图：

1.3 NVIDIA GeForce GTX 1080Ti 显卡驱动

https://www.geforce.cn/drivers
按以下条件筛选：

一般选列表第一个：

点击下载按钮：

1.4 CUDA

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
按以下条筛选并下载：

1.5 cuDNN

https://developer.nvidia.com/rdp/cuDNN-archive
下载这个要向注册一个NVIDIA 开发者账号，微信扫码就行。
按以下条件筛选并下载：

1.6 PyTorch

两个 whl 包：

https://download.pytorch.org/whl/cu92/torch-1.3.0%2Bcu92-cp37-cp37m-linux_x86_64.whl

https://download.pytorch.org/whl/cu92/torchvision-0.4.1%2Bcu92-cp37-cp37m-linux_x86_64.whl

如果要安装其他版本，自行到 https://download.pytorch.org/whl/torch_stable.html 下载。

2. 安装NVIDIA GeForce GTX 1080Ti 显卡驱动（涉及到系统底层修改，请严格按照教程操作，否则后果自负）

2.1 卸载旧驱动（保险起见）

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
sudo apt-get remove --purge nvidia*
./NVIDIA-Linux-x86_64-430.50.run --uninstall

2.2 安装依赖

# 先在软件更新器里面换源再安装
sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic
sudo apt-get install gcc-multilib xorg-dev
sudo apt-get install freeglut3-dev libx11-dev libxmu-dev install libxi-dev  libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

2.3 禁用noueau（系统自带）驱动

# 打开文件 blacklist-nouveau.conf
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf # 图形界面下
sudo vi /etc/modprobe.d/blacklist-nouveau.conf # TTY(终端)下，vim 使用方法自行百度
# 追加以下内容：
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

2.4 禁用 nouveau 内核模块

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u # 更新内核
reboot #重启

2.5 进入 BIOS 关闭 Secure Boot, Fast Boot （具体方法百度）

由于此次配置的机器默认已关闭这两个选项，跳过。

2.6 进入TTY关闭图形界面

重启后，先不要直接登录，必须先进入TTY关闭图形界面装好显卡驱动后才能正常进入系统。
CTRL + ALT + F1，打开TTY，输入用户名，密码，关闭图形界面。

sudo service lightdm stop
sudo service systemctl stop

2.7 运行.run文件安装驱动

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
ls # 获取驱动安装包名称
# 添加权限，文件名自行更改
chmod a+x NVIDIA-Linux-x86_64-430.50.run
# 执行安装包，文件名自行更改，后面的参数不可缺少或错误！
sudo ./NVIDIA-Linux-x86_64-430.50.run --dkms --no-opengl-files

注意执行文件后面的参数 –dkms --no-opengl-files ，一定要输对！
安装包执行后，会提示 pre-script failed ，不用理会，继续安装，安装过程中：
dkms 选 no
32位兼容选 yes
x-org 选 no

2.8 验证安装

nvidia-smi # 若列出GPU的信息列表，表示驱动安装成功

2.9 重启桌面

sudo service lightdm start # 若没自动跳转，CTRL+ALT+F7
# 输入密码，登录系统
nvidia-settings # 若弹出 NVIDIA 设置对话框，亦表示驱动安装成功

3. 安装Anaconda3

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
sudo sh ./Anaconda3-2019.10-Linux-x86_64.sh # 文件名自行更改

输入 yes ，执行安装，安装时各个目录默认即可。

4. 安装Pycharm

4.1 提取安装文件

在文件管理器下载里面找到安装文件，如 pycharm-community-2019.2.3.tar.gz，右键提取。

4.2 执行安装

打开提取出的文件夹（如 pycharm-community-2019.2.3 ），进入其中的 bin 文件夹，右键 打开终端 ，执行以下命令：

sudo sh ./pycharm.sh

然后一直下一步就行，创建快捷方式选是。

4.3 更换解释器

打开安装好的 Pycharm ，依次进入 File -> Settings -> Project *** -> Project Interpreter , 右边小齿轮，Add -> System Interpreter ，右边三个点，选择 Anaconda3 目录下的 python 解释器（ ~/anaconda3/bin/python/python）。

5. 安装CUDA

5.1 执行安装程序

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
sudo sh ./cuda_9.0.176_384.81_linux.run # 文件名自行更改

提示是否安装显卡驱动时，选 no （因为之前已经安装了 NVIDIA 显卡驱动，所以这里就不用重复安装，不要问为什么要提前安装好显卡驱动，问就是必须要这样，再问锤人）。

5.2 添加环境变量

# 打开配置文件
sudo gedit ~/.bashrc
# 在文件末尾追加下面内容：
export PATH=/usr/local/cuda-9.0/bin${PATH:+:$PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# 保存，退出

6. 安装 cuDNN

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
# 解包
sudo dpkg -i libcudnn7_7.6.3.30-1+cuda9.0_amd64.deb # 文件名自行更改
# 安装
sudo apt install libcuDNN7 # 包名自行更改

7. 安装 PyTorch

# 先切换到你下载各个安装包的目录，这里示范的是“下载”目录
cd  ~/下载/
# 安装轮子（安装 whl 的工具）
pip install wheel --user
# 安装 PyTorch 的两个 whl 包，包名自行更改
pip install torch-1.3.0+cu92-cp37-cp37m-linux_x86_64.whl
pip install torchvision-0.4.1+cu92-cp37-cp37m-linux_x86_64.whl

8. 安装 mmdetection

8.1 安装 mmcv （依赖库）

pip install mmcv

8.2 安装 mmdetection

解压包文件 mmdetection-master.zip ，在解压后的文件夹下打开终端，执行以下命令：

python setup.py develop

然后将解压后得到的 mmdet 文件夹复制到Python包文件夹（Python主目录/Lib/site-packages/）下。

8.3 添加 Pycharm 环境变量

打开 Pycharm ，Run -> Edit configurations -> Environment variables，添加以下内容：

LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64/

9. 测试配置是否成功

下载预训练模型（本例中为 faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth ）：
https://github.com/open-mmlab/mmdetection/blob/master/MODEL_ZOO.md
在 Pycharm 中新建一个项目文件夹和一个测试py文件，输入以下内容：

# coding=utf-8from mmdet.apis import init_detector
from mmdet.apis import inference_detector
from mmdet.apis import show_result# 模型配置文件
config_file = '../../configs/faster_rcnn_r50_fpn_1x.py'# 预训练模型文件
checkpoint_file = '../../checkpoints/faster_rcnn_r50_fpn_1x_20181010-3d1b3351.pth'# 通过模型配置文件与预训练文件构建模型
model = init_detector(config_file, checkpoint_file, device='cuda:0')# 测试单张图片并进行展示
img = 'test1.jpg'
result = inference_detector(model, img)
show_result(img, result, model.CLASSES)# 测试一个图像列表并保存结果图像
imgs = ['test1.jpg', 'test2.jpg', 'test3.jpg']
for i, result in enumerate(inference_detector(model, imgs)):show_result(imgs[i], result, model.CLASSES, out_file='result_{}.jpg'.format(i))

运行py文件，执行成功则表示配置成功。
如在配置过程中有任何疑问，欢迎留言！
2019.11.01 23:49 Hafowi

Ubuntu 16.04 配置 PyTorch(GeForce GTX 1080Ti + CUDA 9.0 + Python 3.7) 全套教程相关推荐

linux 1060显卡,Ubuntu 16.04 安装 NVIDIA GeForce GTX 1060 显卡驱动，以及 CUDA 10.1
首先安装nvidia显卡驱动打开终端,先删除旧的驱动: sudo apt-get purge nvidia* 禁用自带的驱动 (很重要!),通过如下命令创建一个文件: sudo vim /etc/ ...
ubuntu 16.04 配置Python2.7 和 Python3.5 同时调用OpenCV
安装OpenCV OpenCV 官网,下载见 SourceForge and GitHub. 若使用官网版本安装不成功,则可试试Github版本. 或者 git clone https://githu ...
linux nginx 配置端口访问,Linux入门教程:ubuntu 16.04配置nginx服务器实现一个IP一个端口多个站点,ubuntunginxNginx 使用异步...
Linux入门教程:ubuntu 16.04配置nginx服务器实现一个IP一个端口多个站点,ubuntunginxNginx 使用异步特点: Nginx 可以部署在网络上使用 FastCGI,脚本 ...
Ubuntu 16.04配置CUDA 9.0+cudnn 7.0以及解决Nvidia显卡导致黑屏问题
Ubuntu 16.04配置CUDA 9.0+cudnn 7.0以及解决Nvidia显卡导致黑屏问题参考文章: (1)Ubuntu 16.04配置CUDA 9.0+cudnn 7.0以及解决Nvid ...
Ubuntu 16.04 配置固定IP地址
1. 查看网卡设备名称 ifconfig 2. 修改network/interfaces sudo vi /etc/network/interfaces vim操作参看Linux(Ubuntu)下Vi ...
worker mpm php,Ubuntu 16.04 安裝 Apache2.4 + mpm_worker + fastcgi + php7.0
之前寫過一篇 Ubuntu 16.04 在 Apache2.4 跑 PHP7.0 是用 mod_php 跑來的,而 mod_php 是使用 mpm_prefork 來跑,這篇主要要說的是 mpm_wo ...
【Ubuntu 16.04中为Chromium、Chrome、Firefox安装Flash播放器插件教程】
Ubuntu 16.04中为Chromium.Chrome.Firefox安装Flash播放器插件教程: Ubuntu 16.04中安装Chromium浏览器 http://www.linuxidc ...
ubuntu 16.04配置嵌入式linux开发环境出现Unable to locate package lib32bz2-1.0
问题描述:Ubuntu16 64位安装android studio的32位库 sudo apt-get install lib32bz2-1.0失败虚拟机上运行ubuntu 16.04 环境 ,打开 ...
Ubuntu 16.04 配置Raft（分布式共识协议）及Maven+Java运行环境
本文用于在Ubuntu 16.04中配置分布式共识协议 Raft, 运行环境为Java + Maven, 开辟端口为 8051-8080, demon中实际使用端口为8051-8053 Raft软件包 ...

Ubuntu 16.04 配置 PyTorch(GeForce GTX 1080Ti + CUDA 9.0 + Python 3.7) 全套教程