ubuntu下conda虚拟环境的操作,cuda,cudnn版本的查询, pytorch的安装

一. ubuntu下conda虚拟环境的操作

随着深度学习的发展,tensorflow, keras, pytorch等深度学习框架的兴起和发展,或者多用户的使用情况,使得在ubuntu下我们可能需要安装多个深度学习框架,为了避免各个深度学习框架之间的影响,可以使用conda创建不同的虚拟环境,方便使用。

在conda下操作虚拟环境有以下几个重要的命令:

(1)查看conda的虚拟环境有哪些:

conda info --envs   

使用上述指令后,看一看到类似以下情况,下图中有两个环境,其中base表示安装好conda(下图中的是miniconda3,和conda几无区别,是一样的)后的环境,pyEnv36表示自己创建的新的虚拟环境 

(2) 新建虚拟环境:

conda create -n pyEnv36 python=3.6

(3) 删除虚拟环境:

conda remove -n pyEnv36 --all

(4) 激活虚拟环境

conda activate pyEnv36

或者

. activate pyEnv36

(5) 退出虚拟环境

conda deactivate pyEnv36

或者

. deactivate pyEnv36

二.  cuda, cudnn版本的查询

在ubuntu下查看cuda的方式如下:

(1)使用命令:

nvcc -V

 (2)或者使用如下命令:

cat /usr/local/cuda/version.txt

在ubuntu下查看cudnn版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

三. pytorch的安装

(1)清华镜像源配置

由于墙的问题,用conda安装Pytorch过程中会连接失败,可以用清华镜像源,将其加入到conda的配置,操作如下:

# 添加Anaconda的清华镜像源
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

# 设置搜索时显示配置通道地址
$ conda config --set show_channel_urls yes

 可以用如下操作查看当前使用源以及删除指定源

conda config --show-sources   # 查看当前使用源

conda config --remove channels 源链接     # 删除指定链接源

(2)开始安装Pytorch

在创建虚拟环境后, 激活虚拟环境,然后再该虚拟环境下安装pytorch,安装前需要知道自己ubuntu系统上安装的cuda的版本。

然后再pytorch的官网:https://pytorch.org/,即可找到安装指令(注意选择合适自己电脑的安装指令):

注:建立虚拟环境是为了在存在多个配置环境时,各个环境之间独立互不影响,如不牵扯该问题,可以不建立虚拟环境,直接安装Pytorch,并且Pytorch,tensorflow和keras可以共存于一个环境中,互不影响。 

(3)测试Pytorch是否安装成功

激活安装Pytorch的虚拟环境后(如无虚拟环境,则不用激活),进入python模式下,看Pytorch能否导入成功以及查看pytorch的版本:以下说明pytorch安装成功。

 

  • 1
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值