Linux配置pytorch

配置pytorch

安装Anaconda

下载Anaconda

wget https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh

安装

bash Anaconda3-5.2.0-Linux-x86_64.sh

环境变量

echo ". /home/ubuntu/anaconda3/etc/profile.d/conda.sh" >> ~/.bashrc

生效配置

source .bashrc

查看是否生效

conda -V

Anaconda创建环境

创建环境

conda create -n pytorch python=3.7

激活环境

conda activate pytorch

Anaconda常用命令

查看已安装

conda list

退出当前环境

conda deactivate

查看conda已安装的环境

conda info --env

Anaconda安装pytorch

下载pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

进入python

python

导入torch库

import torch

查看torch

torch.__version__