开发者
资源

前提条件

在使用TensorFlow GPU2Ascend工具前,您需要完成以下准备工作。

环境准备

  1. 准备一台基于Atlas 训练系列产品的训练服务器,并安装对应的驱动和固件。
  2. 安装开发套件包Ascend-cann-toolkit、框架插件包Ascend-cann-tfplugin和深度学习框架TensorFlow 1.15.0,具体可参考CANN 软件安装指南的“安装开发环境”章节。
  3. 安装MindStudio,具体可参考MindStudio安装指南的“安装操作(Linux)”章节。
  4. 使用TensorFlow GPU2Ascend迁移前须执行如下命令安装依赖,如下命令如果使用非root用户安装,需要在安装命令后加上--user
    pip3 install pandas
    pip3 install xlrd==1.2.0
    pip3 install openpyxl
    pip3 install tkintertable
    pip3 install google_pasta

配置环境变量

  1. 以运行用户登录,在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加以下内容(以非root用户的默认安装路径为例)。
    # Ascend-cann-toolkit环境变量(请根据实际路径修改)
    source ~/Ascend/ascend-toolkit/set_env.sh
    
    # tfplugin环境变量(请根据实际路径修改)
    source ~/Ascend/tfplugin/set_env.sh
  2. 执行:wq!命令保存文件并退出。
  3. 执行source ~/.bashrc命令使其立即生效。

数据准备

执行命令下载resnet50模型脚本。
git clone -b r1.13.0 https://github.com/tensorflow/models.git

为减少不必要的脚本迁移时间,可自行删除models目录下的research、samples和tutorials目录。