1.CUDA安装

1.下载cuda
打开终端输入 :

nvidia-smi

查看支持的cuda版本

(在第一行左边 CUDA Version:11.2看版本,我的是11.2)

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu

如果无法查看,则说明尚未安装nvidia驱动,点击附加驱动,选择对应版本的驱动即可自动下载。

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_02


2.下载安装文件。

这里我的cuda版本下载太高,请参考文章进行cuda安装
Ubuntu18.04+显卡驱动安装+CUDA9.1安装+cdDNN7.0.5安装+Anaconda3安装+tensorflow(GPU)安装)

修改于2021年6月22日
针对cuda11.2版本是否太高,pytorch不支持问题:我安装了pytorch1.7版本,代码可以正常运行。

首先去英伟达官网下载cuda安装包:
https://developer.nvidia.com/cuda-toolkit-archive

按照系统配置选择安装包,据说用deb文件安装容易出错,所以直接下载runfile(后缀为.run)文件来安装,例如我想安装cuda11.2,且我的操作系统为Ubuntu16.04,所以我按照下图选择安装包:

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_03

在终端输入对应代码

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_04

$ wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run 
$ sudo sh cuda_11.2.0_460.27.04_linux.run

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_05

输入accept,回车

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_06


cuda安装包是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install,回车

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_07

3.安装完成后,设置环境变量。

终端输入

gedit ~/.bashrc

在文件最后加入以下代码并保存

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64
export PATH=$PATH:/usr/local/cuda-11.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.2

更新环境变量配置

source ~/.bashrc

至此cuda安装完成,输入nvcc -V(或者nvcc --version)命令查看cuda信息,如果显示下面的文字就说明安装成功了。

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_08

2.cudnn的安装

下载安装文件

按需求下载cudnn的安装文件:
https://developer.nvidia.com/rdp/cudnn-archive

cuda启动缓慢 pytorch cuda 11.2 pytorch_linux_09

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_10


我的系统是 Ubuntu16.04 但是发现对应的版本有两个:Runtime 和 Developer

另外还有个 code samples 。。。。。 什么的文件,该怎么选择呢?

Runtime 和 Developer 版本的区别是:
developer library 包含了在Ubuntu系统上开发深度学习时所需的cuDNN头文件,如果你不需要开发编译任何深度学习程序,而只是将其用于运行某些深度学习应用,那么只下载“runtime library”就足够了。
我是把3个文件都下载下来了,分别安装。

进入下载deb 文件所在的文件,打开终端,然后输入下列三行,安装三个文件

sudo dpkg -i libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-samples_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb

验证是否安装成功:

ldconfig -v | grep cudnn

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_11

如上图,安装成功!

3.Anaconda安装

参照
https://m.linuxidc.com/Linux/2020-02/162304.htm

首先下载anaconda3镜像,清华大学开源软件镜像站下载地址: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 下拉到最低端选择Linux,选择最新版(32/64位)下载。

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_12


cuda启动缓慢 pytorch cuda 11.2 pytorch_linux_13

在下载的文件夹下打开终端,输入

bash Anaconda3-5.2.0-Linux-x86_64.sh

一直点击回车,最后输入yes 回车

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_14


cuda启动缓慢 pytorch cuda 11.2 pytorch_linux_15

继续回车

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_16

输入yes回车

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_17


到这里anaconda已经安装好,接下来可以继续输入yes安装VSCode(我这里一直提示检测网络不能安装,看一些教程说可以不安装,我就没继续下去)

打开新的终端输入下列语句查看anaconda版本信息

anaconda -V
python

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_18

补充于8月28日(参考ubuntu 下使用Anaconda3 出现anaconda:未找到命令

若终端输入anaconda -V,提示anaconda:未找到命令

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_19


则在终端输入

export PATH=~/anaconda3/bin:$PATH 再输入anaconda -V即可

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_20

想要打开anaconda图形界面就打开终端输入

conda install -c anaconda anaconda-navigator
anaconda-navigator

这里不知道为什么我的电脑一直打不开,等解决了再来补充

4.PyTorch环境搭建

1.先创建一个PyTorch虚拟环境
(这里的pytorch是我的环境名,可以改成别的)

conda create -n pytorch python=3.6

激活PyTorch虚拟环境

conda activate pytorch

我这里没激活成功,查了一个方法:

1首先终端输入 source activate

2 然后终端输入 source deactivate

3 输入你要激活的虚拟环境指令 conda activate your_virtual_name

问题解决!!!

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_21


补充于2021年6月22日,因为我重装了一次pytorch,创建环境时一直不成功,记录一下。参考了连接:

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_22


因为我之前弄了清华源镜像,所以创建环境时一直失败,解决办法:
修改其包管理镜像为国内源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

重新新建环境仍旧报错:
解决办法:
打开~/.condarc,删除已有内容,加入如下内容:

channels:
  - defaults
show_channel_urls: true
ssl_verify: false
report_errors: false

成功!

键入“conda env list”,即可查看已创建的虚拟环境列表。

conda env list

2.安装PyTorch库

方法一:下载困难,可自行跳到方法二或新加方法

官网:https://pytorch.org/ 选择对应版本,复制安装命令:

cuda启动缓慢 pytorch cuda 11.2 pytorch_安装包_23

进入环境输入对应口令

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

回车键之后,等一会它会加载出将要安装的一些安装包名称以及大小
其主要三个大包是pytorch、torchvision、cudatoolkit,如果直接安装,几乎是失败的

方法二:
先在清华源上 ,下载出三个安装包

这几个是下载的网站,搭配官网一起使用

1.https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 2.https://download.pytorch.org/whl/cu100/torch_stable.html 3.https://download.pytorch.org/whl/torch_stable.html

我的电脑对应版本的下载链接

https://download.pytorch.org/whl/cu111/torchvision-0.9.0%2Bcu111-cp36-cp36m-linux_x86_64.whl
https://download.pytorch.org/whl/cu110/torchvision-0.8.0-cp36-cp36m-linux_x86_64.whl
https://download.pytorch.org/whl/torchaudio-0.8.0-cp36-cp36m-linux_x86_64.whl

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_24


在虚拟环境下依次输入:

conda activate pytorch 
pip install /home/ubuntu/下载/torchaudio-0.8.0-cp36-cp36m-linux_x86_64.whl
pip install /home/ubuntu/下载/torchvision-0.8.0-cp36-cp36m-linux_x86_64.whl
pip install /home/ubuntu/下载/torchvision-0.9.0+cu111-cp36-cp36m-linux_x86_64.whl

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_25

安装成功!

补充于2021年6月22日,cuda版本11.2,我导师的服务器安装pytorch1.8版本运行代码报错,提示cuda版本过高,但是我配置低版本cuda也报错。所以我选择安装pytorch1.7版本,代码可以正常运行啦~~(配环境挺磨人,一定要有耐心!)
大家可以参考链接安装pytorch

这里我选择的pytorch版本为1.7==(为了快速安装用了清华源,具体操作看上述链接,以下是我用到的终端语句)

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

配置好清华源,去pytorch官网下载所需版本(我的1.7,现在官网首页好像时1.8还是1.9的)。

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_26


cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_27

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0

1.因为在清华源下载,记得把最后的 -c python删除
2.请不要再问我为什么cudatoolkit=11.0,因为这里就这个,我也不懂为啥是这个,我只是小白!帮不上大家提前说声抱歉!

3.测试一下
进入pytorch环境

conda activate pytorch
python
import torch
torch.cuda.is_available()

cuda启动缓慢 pytorch cuda 11.2 pytorch_ubuntu_28

!成功啦!!!
后续补充pycharm安装!

2021年6月22日:ubuntu安装pycharm直接在ubuntu软件里查找就OK,选pycharm CE,我暂时用这个还没什么问题 win10配置pycharm等我有空再整。

cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_29


cuda启动缓慢 pytorch cuda 11.2 pytorch_cuda启动缓慢 pytorch_30

小王出品,必属精品,哈哈哈哈哈