linux 下设置 cuda 的默认环境变量

最新推荐文章于 2024-05-22 20:19:39 发布

sinat_34766791

最新推荐文章于 2024-05-22 20:19:39 发布

阅读量1w

点赞数 1

分类专栏：深度学习文章标签：深度学习

本文链接：https://blog.csdn.net/sinat_34766791/article/details/84458048

版权

在 linux 下使用 pip 安装 tensorlfow 的时候（官方教程：https://www.tensorflow.org/install/pip），如果 cuda 的版本不被 tensorflow 支持，安装不会报错，依然会成功。但是实际无法运行程序，利用官方的测试代码检查：

$ python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

便会出现 import error 的错误。

存在使用源码编译的教程（https://www.tensorflow.org/install/source），可以选择 cuda 版本（例如10.0），但是过程比较繁琐，消耗时间高达3-4小时。因此我选择改变 cuda 的默认环境变量，非常简单，只需要两行代码，然后轻松使用 pip 安装。

我在 linux 系统下同时安装了多个版本的 cuda，默认的环境变量指向最新的 cuda 目录（10.0）.

在安装 tensorflow 的时候，由于 tensorflow 最高支持 cuda-9.0，需要将莫仍环境变量指向 cuda-

关注