anaconda安装tensorflow-gpu注意事项

最新推荐文章于 2024-03-28 20:14:30 发布

原创

最新推荐文章于 2024-03-28 20:14:30 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文详细介绍了如何在Anaconda环境中安装TensorFlow-GPU，包括查询CUDA与Anaconda的Python版本匹配，下载合适的tensorflow-gpu版本，解决安装过程中遇到的问题以及测试GPU设备是否成功启用。

1 查询cuda与anaconda中的python版本（已添加环境变量）
在这里插入图片描述

2 查询cuda和python对应的tensorflow-gpu版本
下载网址：https://www.tensorflow.org/install/pip?lang=python2
我下载的版本：
tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl

3 安装（可以新建虚拟环境安装，也可以在ite-packages中使用pip install安装，本文选择后者）
将下载的tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl拷贝至Anaconda3\Lib\site-packages中，进入该文件夹中，输入命令pip install tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl,出现错误：
在这里插入图片描述
解决方法：
pip install --user tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl

4 测试代码
import tensorflow as tf
print(tf.versi

最低0.47元/天解锁文章