首先安装linux对应版本的Anaconda环境
这里我下载的是
Anaconda3-5.2.0-Linux-x86_64.sh
之后我们只要创建对应的虚拟环境,简单配制即可
create -n pytorch python=3.8
创建了一个python3.8的虚拟环境,再进入虚拟环境
conda activate pytorch
然后在pytorch官网找到安装命令,
conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
下载了pytorch1.8版本,cuda10.2,下载速度还是很快的
安装完成之后测试一下
python
import torch
print(torch.version)
print(torch.cuda.is_available())
转载于作者:杭漂一族小张 https://www.bilibili.com/read/cv15156700/ 出处:bilibili