创建Anaconda虚拟环境
conda create --name mm python=3.8
激活该环境
conda activate mm
安装PyTorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
验证
#python
python 3.8.12 (default, Oct 12 2021, 13:49:34)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>import torch
>>> torch.cuda.is_available()
True
>>> torch.cuda.device_count()
1
退出
>>>quit()
下载包
conda install -n name package
关闭虚拟环境
Linux下:source deactivate
删除虚拟环境
conda remove -n name --all
部分包安装时出现问题,如simpleitk,可用以下命令进行安装
conda install --channel https://conda.anaconda.org/SimpleITK SimpleITK
screen命令
screen -S name # 创建
screen -r name # 恢复
screen -d name # 离线