一. ubuntu下conda虚拟环境的操作
随着深度学习的发展,tensorflow, keras, pytorch等深度学习框架的兴起和发展,或者多用户的使用情况,使得在ubuntu下我们可能需要安装多个深度学习框架,为了避免各个深度学习框架之间的影响,可以使用conda创建不同的虚拟环境,方便使用。
在conda下操作虚拟环境有以下几个重要的命令:
(1)查看conda的虚拟环境有哪些:
conda info --envs
使用上述指令后,看一看到类似以下情况,下图中有两个环境,其中base表示安装好conda(下图中的是miniconda3,和几无区别)后的环境,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的安装
在创建虚拟环境后, 激活虚拟环境,然后再该虚拟环境下安装pytorch,安装前需要知道自己ubuntu系统上安装的cuda的版本。
然后再pytorch的官网:https://pytorch.org/,即可找到安装指令: