ubuntu20.04 安装Anaconda3+CUDA+cudnn+Pytorch

1、安装Anaconda:

打开终端,转到自己的下载目录下(命令为:cd /home/oliver/下载

),然后输入以下命令:

bash Anaconda3-2020.11-Linux-x86_64.sh

终端中查看Anaconda版本:conda -V  或者  conda --version

2、安装NVIDIA驱动:(我选用的方法2)

(方法1)查看可安装的驱动:ubuntu-drivers devices

自动安装推荐的驱动:sudo ubuntu-drivers autoinstall #自动安装推荐驱动,安装完成后会需要定一个密码,一会儿要用,可以设置和开机密码一样

重启电脑:sudo reboot

(方法2)也可以在系统的“软件和更新”下的“附加驱动”里直接选择合适的驱动进行“应用更改”,完成后重启电脑。

查看NVIDIA显卡驱动是否安装成功:nvidia-smi

查看NVIDIA显卡配置选项:nvidia-settings

3、安装CUDA:(我选用的方法2)

官网地址:https://developer.nvidia.com/cuda-toolkit-archive

CUDA最新版下载地址:CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

以前版本的CUDA:CUDA Toolkit Archive | NVIDIA Developer

(方法1)系统默认安装CUDA

sudo apt install nvidia-cuda-toolkit (默认安装)

(方法2)下载CUDA到本地安装

打开终端,转到自己的下载目录下,然后输入以下命令:(本地安装指引)

chmod +x cuda_11.1.0_455.23.05_linux.run(有的博客中没有这一句)

sudo sh cuda_11.1.0_455.23.05_linux.run

(方法3)本地安装(本人没试过)

Installation Instructions:(本地安装指引)

    `sudo dpkg -i cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39_1.0-1_amd64.deb`

    `sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`

    `sudo apt-get update`

    `sudo apt-get install cuda`

添加环境变量,打开profile文件:sudo gedit /etc/profile

在末尾处添加以下内容(注意不要有空格,不然会报错):

export PATH=/usr/local/cuda-11.1/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64$LD_LIBRARY_PATH

安装完,重启之后,查看CUDA版本:nvcc -V  或者  nvcc --version

4、安装对应的cuDNN

官网:NVIDIA cuDNN | NVIDIA Developer

下载cuDNN到本地安装

打开终端,转到自己的下载目录下,然后输入以下命令:(本地安装指引)

sudo dpkg -i libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb

sudo dpkg -i libcudnn8-dev_8.1.1.33-1+cuda11.2_amd64.deb

sudo dpkg -i libcudnn8-samples_8.1.1.33-1+cuda11.2_amd64.deb

5、安装Pytorch:(我选用的方法4,另外三种都没成功)

进入官网选择适合当前机器环境的Pytorch

官网地址:Start Locally | PyTorch

旧版本:Previous PyTorch Versions | PyTorch

(方法1)官网conda安装,打开终端,输入以下命令:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge(这个官方太慢)

(方法2)也可以添加清华镜像源(https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/)下载,打开终端,输入以下命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

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    

配置完成后运行的command命令改为,这里删除了 -c pytorch

conda install pytorch torchvision cudatoolkit=10.2

conda install pytorch torchvision cudatoolkit=11.1

(方法3)官网pip安装:

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

sudo pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

(方法4)本地pip安装pytorch

下载与cudatoolkit=11.1相对应的pytorch torchvision torchaudio 到本地安装,地址:https://download.pytorch.org/whl/torch_stable.html

下载相对应的以下三个文件:

torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl(https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-linux_x86_64.whl)

torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl

torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl(https://download.pytorch.org/whl/torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl)

转到自己的下载目录下,然后输入以下命令:(本地安装指引)

pip install torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl

pip install torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl

pip install torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl

6、测试pytorch是否安装成功

打开终端下依次输入以下命令:

python #显示python版本

import torch

import torchvision

torch.cuda.is_available()

/----------------------------------------/

在主目录下新建文件夹:cd ~

mkdir software

pip install 待安装包名.whl

卸载:

pip uninstall torchvision

pip uninstall torch

/----------------------------------------/

ubuntu20.04安装OpenCV-python

(我选用的方法1-Option 1)

(方法1)从官网安装

去官网:https://pypi.org/project/opencv-python/,查看安装方法:

    Option 1 - Main modules package: pip install opencv-python

    Option 2 - Full package (contains both main modules and contrib/extra modules): pip install opencv-contrib-python (check contrib/extra modules listing from OpenCV documentation)

(方法2)可从默认的Ubuntu 20.04存储库安装OpenCV。 

要安装它,请运行:

sudo apt update

sudo apt install libopencv-dev python3-opencv

上面的命令将安装OpenCV所需的所有软件包。

通过导入cv2模块并打印OpenCV版本来验证安装:

python3 -c "import cv2; print(cv2.__version__)"

在撰写本文时,存储库中的版本为4.2:

4.2.0python