pytorch安装在linux

下载anaconda后，创建虚拟变量，并使用该虚拟变量

```

export PATH=/home/wl/anaconda3/bin:$PATH

conda env list

##conda create -n your_env_name python=X.X（2.7、3.6等）

source activate mytorch

```

查看cuda版本

```

nvidia-smi

```

实验室cuda版本为12.0 在官网下载比12.0低的版本即可

命令行输入该命令

```

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

```

安装后 验证是否可用

```

import torch

print(torch.__version__)

torch.cuda.is_available()

```