conda虚拟环境安装pytorch(gpu版本)纪实_conda虚拟环境下装 gpu pytorch

1.创建虚拟环境

打开cmd或者Anaconda Prompt，创建 python3.8 环境， openmmlab 可以换成自己想要的环境名

conda create --name openmmlab python=3.8 -y
1

使用如下命令查看是否创建成功

conda env list
1

创建成功

2.安装pytorch

（1）激活环境

先使用如下命令激活环境，your_env_name替换为自己的环境名就好，比如我的就是 openmmlab

conda activate your_env_name
1

（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
1

（4）验证安装

我们先 conda list 一下，看看是不是有我们装好的包

没问题

接下来要用 pytorch 看看是不是可用 CUDA，我们需要经历如下几个步骤：

1.在终端直接运行 python

这个很简单了，只要在终端输入一行代码

python
1

2.导入 torch 并验证 cuda 是否可用

import torch
print(torch.cuda.is_available())
1
2

Ture ! 搞定！

ctrl + z 退出 python

（5）退出环境

直接用 conda deactivate 退出当前环境

3.删除环境

如果要删除环境，可使用如下命令

conda env remove -n 环境名
1

over