百度AI studio配置tensorflow环境

最新推荐文章于 2026-03-31 03:10:20 发布

原创

最新推荐文章于 2026-03-31 03:10:20 发布 · 8.9k 阅读

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

之前做了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.0/

3 将下载的cuda安装到上述新建的目录中
!sh cuda_9.0.176_linux-run --silent --toolkit --toolkitpath=$HOME/cuda-9.0

4 下载cudnn，注意这个需要去官网注册账号，事先下载到自己的电脑然后，
  注意版本要对，然后通过新建数据集上传到ai studio的data/目录下，名字尽量短，下载之后是一个tgz格式   的文件，我把名字改为cudnn-9.0.tgz

5 解压4步下载的文件到根目录,解压之后的cudnn文件名默认为cuda，
!tar -zxvf /home/aistudio/data/data25688