【精选】Conda创建虚拟环境并安装Pytorch 您所在的位置:网站首页 anaconda有pytorch吗 【精选】Conda创建虚拟环境并安装Pytorch

【精选】Conda创建虚拟环境并安装Pytorch

2023-10-21 01:32| 来源: 网络整理| 查看: 265

目录 概述创建虚拟环境虚拟环境的激活与切换安装Pytorch进行验证虚拟环境的复制、复现与删除Jupyter notebook的安装与使用

概述

本文主要针对在Anaconda3中创建虚拟环境并安装Pytorch学习框架,Ubuntu和Windows下操作基本一致。

创建虚拟环境

在终端下,创建环境的命令是

conda create -n your_env_name(虚拟环境名称) python==xx(想要创建的虚拟环境的python版本号)

可以根据自己需要修改 例如要创建基于python_3.8的1.7版本的pytorch 在这里插入图片描述 输入y按下enter确认开始下载安装。 在这里插入图片描述 终端提示上述信息说明虚拟环境创建成功。

虚拟环境的激活与切换

虚拟环境创建完成之后,终端输入以下命令即可激活虚拟环境。

conda activate your_env_name(虚拟环境名称)

退出当前虚拟环境命令如下。

conda deactivate

运行效果如图: 在这里插入图片描述 如果需要切换虚拟环境,建议先退出当前虚拟环境,再进入需要切换到的环境。 否则就会像下图我尝试的这样,发生虚拟环境的嵌套,容易误操作。 在这里插入图片描述 此外,还可以用"conda info -e"和"conda env list"来查看当前已经安装的虚拟环境信息及位置。

conda info -e conda env list

运行效果如图: 在这里插入图片描述

安装Pytorch

激活环境之后,就可以在环境中安装pytorch框架。 进入Pytorch官网点击install进行选择。 在这里插入图片描述 如上图,如果需要安装的是pytorch_1.7版本,可以直接选择系统、安装方式和CUDA版本,复制下方红色框中的命令到终端即可安装。 需要其他版本pytorch可以点击图中绿色框内的选项,查看pytorch的早期版本。 在这里插入图片描述 例如pytorch1.4,复制相应的命令到终端即可安装。

conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=9.2 -c pytorch conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

可根据需要修改安装版本,但需要提前查询兼容和版本对应关系。 另外安装的时候要将命令后的-c pytorch删除,从国内源进行下载,速度快一些。

注意:目前发现删除-c pytorch之后conda无法找到torchvision0.5.0,安装时可删除“torchvision==0.5.0”,待框架安装完毕在自己创建的虚拟环境下用pip安装即可。 conda安装pytorch可以直接复制命令到终端执行。 以下以pytorch_1.7、CUDA11.0为例。 在这里插入图片描述 查看conda找到将要安装的包是否是要进行安装的,如果是,输入y之后enter确认进行下载; 如果不是,就需要重新检查安装pytorch的命令。 在这里插入图片描述

终端提示上述信息说明pytorch框架安装成功。

进行验证

在自己创建的虚拟环境中,依次执行以下代码进行验证:

python import torch torch.cuda.is_available()

如下图提示True说明框架配置成功,且GPU可用。 在这里插入图片描述

虚拟环境的复制、复现与删除

如果工作站已经配置好了某一环境,但出于互不干扰的考虑,需要另外创建一个各软件、库均相同的版本,可以直接利用conda的复制命令。

例如我想复制xwb创建的python3.6和pytorch1.4的虚拟环境,可以直接在终端输入

conda create -n your_env_name(要创建的虚拟环境名称) --clone env_name(要复制的虚拟环境名称)

进行虚拟环境的复制。 注意:复制环境时一定要提前更改名字,以免导致环境错乱。 复制完成之后激活进入使用。 在这里插入图片描述 需要注意的是conda复制时只会将原来环境中用conda install等命令安装的包进行复制,不能够复制pip等命令安装的包和软件。

如果想要导出配置好的虚拟环境,并在另外一台设备上进行虚拟环境的复现。 可以激活进入需要导出复现的虚拟环境,然后在终端执行

conda env export > environment.yaml

导出已有环境。 执行后会在根目录生成一个environment.yaml的文件。 在这里插入图片描述 复制该文件到另外一台设备的根目录,在终端执行“conda env create -f environment.yaml”命令即可复现虚拟环境。 在这里插入图片描述 注意:以上复现虚拟环境是我在win10中进行的尝试,不同操作系统的设备不能相互复现。

而如果觉得某一个虚拟环境多余,就可以直接对其进行删除操作(不可逆,请谨慎操作)

conda remove -n your_env_name(虚拟环境名称) --all 删除虚拟环境 Jupyter notebook的安装与使用

每一个虚拟环境都需要安装jupyter notebook。

conda install jupyter notebook

在这里插入图片描述 在这里插入图片描述 终端提示上述信息说明juputer notebook安装成功。

在利用Xshell或者VSCode远程连接工作站之后,Xshell和VSCode的界面显示分别如下: 在这里插入图片描述 在这里插入图片描述

此时如果要使用jupyter notebook调试程序,需要先激活进入到自己的环境,并进入自己的文件夹下,以确保jupyter的根目录是自己的文件夹。之后再启动jupyter即可。

conda activate your_env_name(虚拟环境名称) cd /data/students/master/****/your_folder(你的文件夹)

注意:一定要激活进入到自己的虚拟环境之后再进行包的安装等操作。



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有