conda里面有没有cuda,Anaconda中使用不同版本的cuda

  • Post author:
  • Post category:其他


使⽤用指南

安装 Anaconda3

检查是否已经安装anaconda,看 which conda 是否有输出 若没有,按照 https://docs.anaconda.com/anaconda/install/进⾏行行安装

创建anaonda环境

比如现在要使⽤用tensorflow1.13,它要求⽤用cuda 10.0.

第⼀步 创建环境

conda create -n tf113_cuda10 python=3.6

环境名为 tf113_cuda10 ,指定python版本为 3.6 (tensorflow尚不不⽀支持3.7及以上的版本)

第⼆步 指定该环境使⽤用的cuda版本

tensorflow运⾏行行时,实际是从动态链接库的地址找需要的cuda动态链接库,所以我们只需要指定 LD_LIBRARY_PATH . 即保证该环境启动时,把cuda 10的地址加⼊入 LD_LIBRARY_PATH ,关闭该环境时

再移除.

创建环境启动时的脚本:

mkdir -p ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d

touch ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh

vim ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh

chmod +x ~/anaconda3/envs/tf113_cuda10/etc/conda/activate.d/activate.sh

在脚本中填⼊入:

#!/bin/sh

ORIGINAL_LD_LIBRARY_