前提条件
在使用PyTorch GPU2Ascend工具前，您需要完成以下准备工作。
环境准备
- 准备一台基于Atlas 训练系列产品的训练服务器，并安装对应的驱动和固件。
- 安装开发套件包Ascend-cann-toolkit，具体可参考《CANN 软件安装指南》的“安装开发环境”章节。
- 安装MindStudio，具体可参考《MindStudio安装指南》的“安装操作（Linux）”章节。
- 安装PyTorch 1.11.0，具体可参考《CANN 软件安装指南》>安装PyTorch。
- 使用PyTorch GPU2Ascend迁移前须执行如下命令安装依赖，如下命令如果使用非root用户安装，需要在安装命令后加上--user。
pip3 install pandas #pandas版本号需大于或等于1.2.4 pip3 install libcst #语义分析库，用于解析Python文件 pip3 install prettytable #将数据可视化为图表形式
配置环境变量
- 以运行用户登录，在任意目录下执行vi ~/.bashrc命令，打开.bashrc文件，在文件最后一行后面添加以下内容（以非root用户的默认安装路径为例）。
# Ascend-cann-toolkit环境变量（请根据实际路径修改） source ~/Ascend/ascend-toolkit/set_env.sh # PyTorch环境变量（请根据PyTorch所在实际路径修改） export LD_LIBRARY_PATH=~/.local/lib/python3.7/site-packages/torch/lib:$LD_LIBRARY_PATH
- 执行:wq!命令保存文件并退出。
- 执行source ~/.bashrc命令使其立即生效。