Conda建立虚拟环境及torch安装

本人的系统环境:

Ubuntu 16.04
Cuda 9.0 + Cudnn 7.0.5
Python 3.7
Anaconda 3 
这里推荐大家使用Anaconda,可以比较方便的创建Python虚拟环境,避免不同的Python库之间产生冲突。

一 anaconda安装


         anaconda的安装非常简单,可以选择去官网下载,但是因为anaconda的服务器在国外,所以国内的我们在Ubuntu下用pip或者conda安装软件包都容易龟速最终导致失败,但是嗑盐的我们耽误不起时间,所以我们要找一个最好最快的安装的方式,使用清华大学开源软件镜像站 ,下拉到最底端选择Linux,选择最新版(32/64位)下载。

下载完成后,使用bash运行sh脚本文件

bash Anaconda3-5.3.1-Linux-x86_64.sh

 接下来,按照提示进行安装

按住回车键一直到进度为100%,阅读完注册信息后,输入“yes”回车继续; 

按回车键确认安装位置 /home/{虚拟机名称}/anaconda3 ;

输入“yes”选择添加环境变量; 

输入“no”取消安装 VSCode 

完成以上步骤之后,anaconda安装成功,可以在终端里输入Python查看一下,因为anaconda的Python会替代Ubuntu的默认python 

 二 conda创建、查看、删除虚拟环境 

1   anaconda创建虚拟环境,创建名称为mmdetection的Python3.7环境

conda create -n mmdetection python=3.7

2  激活虚拟环境,环境名称和上面一致

source activate mmdetection

3  退出虚拟环境

source deactivate mmdetection

4  删除环境,删除之前必须退出虚拟环境 

conda remove -n mmdetection --all

如果建立虚拟环境较多,可以使用下面命令查看:

conda-env list

三   conda安装pytorch

pytorch最近已经更新到了稳定版本的1.1,从Pytorch官网上可以看到有多种安装方式:

比较常见的方法是通过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤其是文件:pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2,下载的时候可能会遇到无尽的等待。这里推荐用清华源替代默认conda源的方法,可以解决下载过慢的问题。

清华conda源地址:anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

1 添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

2 安装pytorch

根据Python和CUDA选择对应的版本,然后官方给出提示可通过运行:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
但是这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:
 

conda install pytorch torchvision cudatoolkit=9.0

 安装完成之后,可以进行检查是否安装成功

reference

conda安装Pytorch下载过慢解决办法(3月5日更新ubuntu下pytorch1.0.1安装方法)

  • 21
    点赞
  • 114
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值