创建虚拟环境
conda create -n env_name python=X.X(python版本号)
一定要说明Python版本号,否则安装好会缺少很多Python相关的依赖包。
查看虚拟环境
conda env list
如图可以看到当前的虚拟环境
激活虚拟环境
conda activate env_name
退出当前虚拟环境
conda deactivate
配置pytorch
安装前记得看一下服务器的cuda版本
nvidia-smi
我的是11.4,然后进入pytorch官网,选择对应的下载版本,下载pytorch时选择小于或等于cuda版本,避免兼容问题。我选择了历史版本10.2。
结果发生报错后来选择了pip安装,安装成功。
pip install torch==1.12.0+cu102 torchvision==0.13.0+cu102 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu102
检测是否安装成功输入python
然后输入import torch
再输入torch.cuda.is_available(),返回结果为true,则说明安装成功。
Python从命令行中退出:
1、esc+enter
2、Ctrl+Z+enter