1.创建虚拟环境
打开cmd或者Anaconda Prompt,创建 python3.8 环境, openmmlab 可以换成自己想要的环境名
conda create --name openmmlab python=3.8 -y
使用如下命令查看是否创建成功
conda env list
创建成功
2.安装pytorch
(1)激活环境
先使用如下命令激活环境,your_env_name替换为自己的环境名就好,比如我的就是 openmmlab
conda activate your_env_name
(2)下载对应版本Pytorch
激活环境后,在终端输入(版本可以换成自己想要的,而且注意torch,torchvision,torchaudio三个之间有版本对应关系):
python -m pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
(4)验证安装
我们先 conda list
一下,看看是不是有我们装好的包
没问题
接下来要用 pytorch 看看是不是可用 CUDA,我们需要经历如下几个步骤:
1.在终端直接运行 python
这个很简单了,只要在终端输入一行代码
python
2.导入 torch 并验证 cuda 是否可用
import torch
print(torch.cuda.is_available())
Ture ! 搞定!
ctrl + z
退出 python
(5)退出环境
直接用 conda deactivate
退出当前环境
3.删除环境
如果要删除环境,可使用如下命令
conda env remove -n 环境名