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

一. 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/,即可找到安装指令:

猜你喜欢

转载自blog.csdn.net/yangzhengzheng95/article/details/86222094