linux系统 在python3.6/CUDA 11环境下安装tensorflow 1.15

今天在实验室服务器(3090)上跑别人用tensorflow写的代码,CPU使用率飙高,吓得我赶紧停了QAQ。
后来发现是因为GPU无法使用,其原因是官网中cuda11.X,仅支持tf2.X,不支持tf1.X。
通过查阅资料,参考大佬的方法,最终成功在cuda11、python3.6环境下使用tf-gpu 1.15版本:)

参考网址:https://www.it610.com/article/1419985725641838592.htm

具体方法如下:

# 推荐在虚拟环境下操作(非必要)
# python3 -m virtualenv venv
# source venv/bin/activate
pip install --upgrade pip
pip install nvidia-pyindex
pip install nvidia-tensorflow[horovod]
pip install nvidia-tensorboard==1.15

查看tensorflow-gpu是否可用:

import tensorflow as tf
print(tf.test.is_gpu_available())	# 为True则成功

测试方法:

import tensorflow as tf
import tensorboard

tf.enable_eager_execution()
a = tf.random.uniform([1000, 1000])
b = tf.random.uniform([1000, 1000])
tf.matmul(a, b)		# 检查输出即可
### TensorFlow 1.15 兼容的 Python 版本及安装解决方案 TensorFlow 1.15 是一个较老但仍然被广泛使用的版本,尤其在一些需要长期支持(LTS)或特定硬件兼容性的项目中。为了确保与 TensorFlow 1.15 的最佳兼容性,推荐使用 Python 3.5 至 3.8 的范围[^2]。然而,根据官方文档和社区反馈,Python 3.63.7 是最稳定的选项[^3]。 #### 环境创建与配置 在使用 Conda 创建环境时,可以明确指定 Python 版本以避免潜在的兼容性问题。以下是一个典型的命令,用于创建一个包含 Python 3.6 的新环境,并安装 TensorFlow 1.15: ```bash conda create -n tf115_env python=3.6 conda activate tf115_env pip install tensorflow==1.15 ``` 通过上述步骤,可以确保环境中的 Python 版本与 TensorFlow 1.15 完全兼容[^4]。 #### 常见安装问题及解决方法 1. **依赖冲突**:如果系统中已存在其他版本的 TensorFlow 或相关库,可能会导致安装失败。建议在独立的 Conda 环境中进行安装。 2. **CUDA 和 cuDNN 配置**:对于 GPU 支持,TensorFlow 1.15 要求 CUDA 10.0 和 cuDNN 7.4.2 或更高版本[^5]。确保这些库已正确安装并配置路径。 3. **pip 缓存问题**:有时 pip 缓存可能导致旧版本包被错误安装。可以通过添加 `--no-cache-dir` 参数强制忽略缓存: ```bash pip install tensorflow==1.15 --no-cache-dir ``` #### 验证安装 安装完成后,可以通过以下代码验证 TensorFlow 是否正常工作: ```python import tensorflow as tf print(tf.__version__) hello = tf.constant("Hello, TensorFlow!") sess = tf.Session() print(sess.run(hello)) ``` 如果输出为 `Hello, TensorFlow!` 并显示版本号为 `1.15.0`,则说明安装成功[^6]。
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值