百度AI studio配置tensorflow环境

最新推荐文章于 2025-03-21 15:55:46 发布

firesodier

最新推荐文章于 2025-03-21 15:55:46 发布

阅读量8.7k

点赞数 8

分类专栏： tensorflow

本文链接：https://blog.csdn.net/firesolider/article/details/105023062

版权

本文档详细介绍了如何在百度AI Studio上配置TensorFlow环境，包括安装准备、新建和激活虚拟环境，以及自动化安装流程。特别强调了环境版本匹配的重要性，提供了自定义安装TensorFlow版本的方法，并给出了自动化脚本env.sh的示例，确保每次启动都能快速激活所需环境。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

之前做了ai studio 安装pytorch环境的教程后来发现tensorflow也是可以安装的。由于ai studio默认python环境只有2.7和3.7

有些tensorflow版本是装不了的,注意以“！”开头的脚本可以在notebook执行，没加的最好在终端执行

我这里的教程可以自己选择安装任何版本的tensorflow，

安装之前注意自己的python版本，tensorflow版本，cuda版本，cudnn版本，一定要相互匹配

下面给出我的作为参考

python 3.6
tensorflow 1.12.0
cuda 9.0
cudnn 7.4.1
Linux 16.04 (ai studio系统版本)

1 安装准备

1 在ai studio或者notebook下载cuda，注意把下载之后的cuda安装文件的名字改短一点
   这里我改为cuda_9.0.176_linux-run
!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run

2 新建一个目录cuda-9