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