1、安装显卡驱动;
2、安装CUDA;
3、安装CUDNN;
4、安装Conda;
5、安装Pytorch。
1、Ubuntu 20.04
2、显卡:1050Ti
(已经安装的可以跳过)
选择一个推荐的驱动,安装即可。
sudo ubuntu-drivers autoinstall
手动安装之前要知道自己的显卡型号和推荐的驱动型号。
查看显卡型号:
ubuntu-drivers devices
手动安装驱动(以1050Ti为例)
sudo apt install nvidia-driver-515
在终端输入
nvidia-smi
出现如图内容,则驱动安装成功
首先是CUDA和驱动的对应关系,可以从此网站查阅;
确定好版本之后,从此网站下载。
我个人而言一般会按照高版本的驱动+低版本的CUDA+与CUDA对应的cudnn+与CUDA对应的PyTorch这样的顺序去配置环境。
——土豆
本文以下载CUDA10.2为例
Ubuntu20.04自带的GCC版本为9.7,要安装CUDA10.2需要降级为7
输入命令安装gcc7
apt-get install gcc-7 g++-7
查看gcc版本,可以看到目前系统中存在7和9两个版本
ls /usr/bin/gcc*
ls /usr/bin/g++*
使用update-alternatives进行版本切换(修改优先级),输入以下命令
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 50
查看当前gcc和g++版本
sudo update-alternatives --config gcc
sudo update-alternatives --config g++
可以看到优先级最高的是gcc7和g++7。
转到下载CUDA的路径,先给权限,在执行
sudo chmod a+x cuda_10.2.89_440.33.01_linux.run
sudo ./cuda_10.2.89_440.33.01_linux.run
选择continue-->输入accept-->按空格去掉安装显卡驱动的选项-->选择install
sudo nano ~/.bashrc
在尾部加入以下内容(路径不一定和我一样,看版本号)
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
更新环境变量
source ~/.bashrc
查看CUDA版本信息
nvcc -V
进入samples,(看当时安装的时候,案例放在了哪里)
cd /usr/local/cuda-10.2/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
出现以下结果说明安装成功
cd /usr/local/cuda-10.2/samples/1_Utilities/bandwidthTest
sudo make
./bandwidthTest
结果为PASS即为通过。
下载与CUDA版本对应的cudnn,下载地址(注册之后才能下载)
本文以CUDA10.2和1050Ti为例
下载Library for Linux, Ubuntu(x86_64 & PPC architecture)中的,最好下载到一个地方
cuDNN Library for Linux (x86)
cuDNN Runtime Library for Ubuntu18.04 (Deb)
cuDNN Developer Library for Ubuntu18.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)
解压
tar zxvf cudnn-10.2-linux-x64-v8.0.5.39.tgz
进行文件替换(注意路径)
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
加权限(注意路径)
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安装其余三个deb包
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb
测试环节
cd /usr/src/cudnn_samples_v8/mnistCUDNN
sudo make
./mnistCUDNN
出现下面内容
本文以miniconda为例,下载地址,对应好自己的系统版本和想用的python版本
本文以下面配置为例:
Python 3.8 | Miniconda3 Linux 64-bit |
下载完成之后,进入相应目录,运行
bash ./Miniconda3-py38_4.12.0-Linux-x86_64.sh
1、安装过程中根据提示输入enter;
2、选择安装路径的时候可以默认或者是自定义,因为默认的路径root不一定有权限,可以放在opt里面,例如路径为/opt/miniconda3;
3、最后一部问是否添加环境变量,最好选择yes;
4、验证安装(重开终端窗口)(如果不行可能是环境变量没加进去)(或source环境)
conda -V
查看当前所有的env
conda env list
创建新的env
conda create -n pt python=3.8
例如这个语句就是创建一个名字叫pt的,python版本为3.8的一个虚拟环境。
激活环境命令
conda activate pt
可看到终端命令前面出现了(pt),说明已经在环境pt里面
退出环境命令
conda deactivate
首先要进入自己的conda虚拟环境,终端前面带(pt)或其它。
安装链接在这里
https://pytorch.org/get-started/previous-versions/
进入之后,可以看到不同操作系统,不同conda,不同cuda的命令,根据自己的实际所需选择
以torch1.7.1为例
与自己的cuda版本对应
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch
静待安装完成就ok。
官网地址在这里
下载解压之后,进入解压目录,进入bin目录,./pycharm.sh可以运行,可以在里面设置快捷方式;
在工具栏Tools,有Create Desktop Entry..,可以创建快捷方式。
新建工程,选择
选择环境使用conda,然后Interpreter选择自己刚才建立的虚拟环境
代码测试torch
import torch
print(torch.cuda.is_available())
a = torch.zeros(5, 3)
print(a)
运行之后
则说明pytorch以及搭建完成。
参考链接:
1、NVIDIA三件套环境配置 - PleasureReceiver
2、安装Ubuntu20.04后的配置 - 知乎