TensorFlow GPU版安装方法

首先下载CUDA https://developer.nvidia.com/computeworks不推荐最新版本在这里插入图片描述
下载完成后,选择路径并安装
在这里插入图片描述
安装完成后下载https://developer.nvidia.com/cudnn对应的cudnn
将cudn解压后得到的三个文件复制到CUDA的安装文件目录下
在这里插入图片描述
至此显卡运算平台搭建完成,接下来就是Python和TensorFlow的安装
PS:Python和TensorFlow的版本是否匹配。
安装Anaconda(https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
在这里插入图片描述
完成后打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入命令:

Conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

创建运行环境,输入命令:

conda create -n tensorflow-gpu python=3.6

新建一个名字叫“tensorflow-gpu”的运行环境。
输入命令:

conda activate tensorflow-gpu 

激活并进入环境

安装tensorflow1.9,输入指令:

conda activate tensorflow-gpu

至此,tensorflow的安装完成。
如果能运行下面的代码并且能看见一些显卡的信息就表示安装成功

import tensorflow as tf
hello = tf.constant('Hello,TensorFlow!')
sess=tf.Session()
print(sess.run(hello))

End

### 正确配置 TensorFlow GPU 本中的 NumPy 依赖 当安装 TensorFlowGPU 本时,确保兼容的 NumPy 本对于系统的稳定性和性能至关重要。通过特定的方法可以有效管理这些依赖关系。 #### 使用 Conda 环境安装 Conda 是一种跨平台的包管理和环境管理系统,能够很好地解决 Python 库之间的本冲突问题。为了使 TensorFlow 和 NumPy 协同工作,在创建新的 conda 虚拟环境中指定所需的 TensorFlow 及其依赖项本是一个明智的选择: ```bash conda create -n tf_gpu python=3.8 conda activate tf_gpu conda install tensorflow-gpu==2.6 numpy=1.19.5 h5py>=3.1.0 ``` 上述命令序列首先创建了一个名为 `tf_gpu` 的新虚拟环境并激活它,接着指定了要安装的具体软件包及其本号[^1]。 #### 利用 Pip 进行安装 如果偏好使用 pip 工具来安装 TensorFlow,则需注意 pypi 自动处理部分依赖关系可能导致某些情况下出现不兼容现象。因此建议先单独安装适当本的 NumPy 后再继续安装 TensorFlow-GPU: ```bash pip install numpy==1.19.5 scipy==1.6.2 --user pip install tensorflow-gpu==2.6 --user ``` 此方法可防止因默认安装最新 NumPy 或其他库而引发潜在的问题[^3]。 #### 验证安装成功与否 完成以上步骤之后,可通过导入模块测试是否一切正常运行: ```python import tensorflow as tf print(tf.reduce_sum(tf.random.normal([1000, 1000]))) ``` 这段简单的代码片段用于确认 TensorFlow 是否已正确加载以及能否利用 GPU 加速计算过程[^4]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值