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

JXNU_SONG

已于 2023-11-27 20:27:30 修改

阅读量4.2k

点赞数 4

分类专栏： python 环境配置文章标签： pytorch conda python

于 2022-11-09 13:53:44 首次发布

本文链接：https://blog.csdn.net/qq_45863710/article/details/127767303

版权

1 篇文章 0 订阅

订阅专栏

1.创建虚拟环境

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

conda create --name openmmlab python=3.8 -y

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

conda env list

创建成功

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

conda activate your_env_name

在这里插入图片描述

激活环境后，在终端输入（版本可以换成自己想要的，而且注意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

我们先 conda list 一下，看看是不是有我们装好的包
在这里插入图片描述

没问题

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

1.在终端直接运行 python

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

python

在这里插入图片描述
2.导入 torch 并验证 cuda 是否可用

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

在这里插入图片描述
Ture ! 搞定！

ctrl + z 退出 python
在这里插入图片描述

直接用 conda deactivate 退出当前环境
在这里插入图片描述

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

conda env remove -n 环境名

关注

专栏目录