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版本
在左下角查看以往版本,previous version of pytorch
根据个人情况选择,复制粘贴即可
5.验证是否安装成功
python
import torch
torch.cuda.is_available()
为True则安装成功