一. 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安装成功。