当前位置: 代码迷 >> 综合 >> Ubuntu18.04 配置深度学习环境
  详细解决方案

Ubuntu18.04 配置深度学习环境

热度:111   发布时间:2023-11-13 23:37:38.0

Ubuntu18.04 配置深度学习环境

运行环境: Ubuntu 18.04

安装完系统,先更换镜像源(ustc),然后apt-get update

安装显卡驱动+cuda

新版本的nvidia-driver安装同时自动可以安装对应最新版本的cuda

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update

查看当前系统推荐你安装的驱动版本

$ ubuntu-drivers devices

安装最新版本

$ ubuntu-drivers devices

重启,查看是否安装成功,发现cuda10.1也被同时自动安装

$ nvidia-smi
Tue May  7 21:05:41 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.09       Driver Version: 430.09       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce RTX 2070    Off  | 00000000:01:00.0 Off |                  N/A |
|  0%   39C    P8    24W / 185W |    218MiB /  7982MiB |      4%      Default |
+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1280      G   /usr/lib/xorg/Xorg                           136MiB |
|    0      1449      G   /usr/bin/gnome-shell                          80MiB |
+-----------------------------------------------------------------------------+

安装cuDNN

官网https://developer.nvidia.com/rdp/cudnn-download 下载,需要注册下账号,选对应cuda的版本的cuDNN,

下载对应的deb包:cuDNN Runtime Library for Ubuntu18.04 (Deb)

得到:libcudnn7_7.5.1.10-1+cuda10.1_amd64.deb

安装即可:

$ sudo dpkg -i libcudnn7_7.5.1.10-1+cuda10.1_amd64.deb

安装相关库

$ sudo apt-get install python3-pip python3-tk
$ sudo pip3 install numpy scipy matplotlib pillow

安装pytorch:最好从官网安装https://pytorch.org/get-started/locally/

选择对应版本的安装命令:

例如torch1.0.1版本,cuda10.X, python3.6, linux系统(目前最新1.1.0版本):

$ sudo pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-linux_x86_64.whl
$ sudo pip3 install torchvision