在Ubuntu中安装Anaconda可以根据 这篇博客,写的十分详细。
装好之后,可以通过通过命令安装版本为X.X(如2.7,3.6)的虚拟环境:
conda create -n your_env_name python=X.X
- 1
其中(your_env_name)是你所创建的虚拟环境的名字。
然后 使用下方命令激活你刚才建立的环境
source activate your_env_name
- 1
激活了环境之后,我们可以看到命令行之前多了一个括号,括号中包含了你的环境名字,这就说明激活成功。
在激活的环境中使用如下命令安装CUDA:
conda install cudatoolkit=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
- 1
我装CUDA的同时,自动装上了CUDNN。
如果你没有装上,也可以使用下方命令安装CUDNN:
conda install cudnn=7.0.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
- 1
其中CUDA的版本可以自己更改,我改成了9.2.0,CUDNN应该也可以改,不过我没有尝试。
这篇文章还展示了Tensorflow不同版本要求与CUDA及CUDNN版本对应关系,一般我们都是装好了tensorflow-gpu,然后需要对应版本的CUDA及CUDNN,可以让大家参考一下。
仍然是在激活的环境中安装tensorflow,比如我的CUDA是9.2.0,我的CUDNN是7.2.1,那么对应的Tensorflow就可以安装1.5.0,如下方命令所示:
pip install tensorflow-gpu==1.5.0
- 1
以上所安装的Python、CUDA、CUDNN以及Tensorflow都是在虚拟环境中安装的,下次想要使用该环境时或者在该环境安装其他库包文件,需要重新进入虚拟环境。与之前的代码一样,如下:
source activate your_env_name
- 1
查询当前环境下的库的版本号可使用以下命令:
conda list cudnn
conda list cuda
- 1
- 2
评论记录:
回复评论: