使用Anaconda安装pytorch

使用Anaconda安装

1，安装Anaconda。

2，创建环境

# -n 后面的是环境的名字，自定义
conda create -n pytorch python=3.8  

3，查看环境

conda env list

4，进入pytorch环境

conda activate pytorch

5，查看环境已安装的库

conda list

6，安装 pytorch（是否有支持cuda）

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

7，验证

import torch
print(torch.cuda.is_available())

8，退出环境

conda deactivate

9，查看环境信息

conda info -e