【安装Pytorch】

1.查看已有的虚拟环境

 conda env list

 

2.创建自己的虚拟环境

 conda create -n name python==3.8
 示例：
 conda create -n pytorch python==3.8

param： name 3.8

pytorch为自己创建的环境名称，随意更换

3.8为python版本

3.激活自己的环境

 conda activate pytorch

激活后 base-pytorch

pytorch是我的环境名

关闭环境代码

 

 conda deactivate

4.查看cuda版本

 nvidia-smi

 

表格右上角 CUDA Version：11.2

根据cuda版本安装对应的pytorch版本

网址：PyTorch

在左下角查看以往版本，previous version of pytorch

 

根据个人情况选择，复制粘贴即可

 

 

5.验证是否安装成功

 python
 import torch

 torch.cuda.is_available()
 为True则安装成功