torch1.8安装教程

1 conda安装

1.1 使用wget下载

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

1.2 执行安装命令

sh Anaconda3-2019.10-Linux-x86_64.sh

1.3 配置~/.bashrc文件并更新

在文件末尾添加

export PATH='anaconda的安装目录/bin:$PATH'

更新文件

source ~/.bashrc

2 Torch-1.8 安装

2.1 创建python3.6环境

conda create -n pytorch1.8 python=3.6

2.2 安装torch1.8

pip install torch -f https://download.pytorch.org/whl/rocm4.0.1/torch_stable.html
pip install ninja && pip3 install ‘git+https://github.com/pytorch/vision.git@v0.9.1’

3 测试

4 cuda可用性测试