【Linux】Ubuntu20.04版本配置pytorch环境2023.09.05【教程】

【Linux】Ubuntu20.04版本配置pytorch环境2023.09.05【教程】

一、安装Anaconda虚拟环境管理器

首先进入Anaconda官网下载linux版本的安装文件Anaconda3-2023.07-2-Linux-x86_64.sh，进入安装文件路径，运行下面的脚本进行安装

bash Anaconda3-2023.07-2-Linux-x86_64.sh

安装需要阅读用户协议，一直按enter就行了，到当前页面时需要输出yes进行确认即可开始安装。

然后按照提示进行操作即可完成安装。

二、安装Miniconda虚拟环境管理器

如果觉得，Anaconda下载太慢的话，可以使用miniconda来代替，

sudo apt update
sudo apt upgrade
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

然后和安装Anaconda类似的执行

bash ./Miniconda3-latest-Linux-x86_64.sh

然后进行安装即可

三、创建虚拟环境并激活

Anaconda基本环境管理语法如下

• 创建虚拟环境

conda create -n <your-virtualenv-name> python=3.8

• 激活虚拟环境

conda activate <your-virtualenv-name>

• 取消激活虚拟环境

deactivate

• 移除虚拟环境

conda env remove -n <your-virtualenv-name>

• 查看所有虚拟环境

conda env list

• 克隆现有的环境

conda create --name <new_env_name> --clone <existing_env_name>

• 卸载虚拟环境中的某个包

conda remove --name <your-virtualenv-name> <pkg-name>

conda uninstall --name <your-virtualenv-name> <pkg-name>

安装完成之后，可以创建虚拟环境，创建虚拟环境完毕之后我们先激活该环境。比如我创建了一个虚拟环境metaRL，然后使用conda activate metaRL进行激活

四、anaconda取消默认激活虚拟环境

如果不想每次启动终端的的时候都激活anaconda虚拟环境，我们可以取消默认激活，打开终端输入以下的语句即可

conda config --set auto_activate_base false

五、安装Pytorch

首先进入Pytorch官网，然后选择您设备的型号，复制相应的安装语句进行安装即可

比如我这里复制的安装语句是

pip3 install torch torchvision torchaudio

然后等待安装结束

经过漫长的等待

终于安装结束了，出现成功安装Suceessfully built lit的提示

六、测试pytorch

安装成功后我们需要测试是否能正常使用，打开终端，在终端中启动python

python

导入pytorch

import torch

查看torch版本

print(torch.__version__)

查看cuda是否可用

print(torch.cuda.is_available())

结果如下，则表示安装成功

如果配置了cuda和cudnn可以使用以下的语句进行查看，具体如何配置请参考这篇：ubuntu配置多版本cuda+cudnn环境，及版本切换方法

torch.version.cuda
torch.backends.cudnn.version()

如下所示

Reference

https://blog.51cto.com/u_16099166/6739522