multi_gpu_test
在多GPU机器上使用PyTorch进行并行化(已在Google Cloud GPU机器上测试)
设置带有一个或多个GPU的Google云机的脚本(要购买GPU机,请参阅以获取说明)
Pytorch安装步骤/脚本
测试实用程序以检查多GPU执行
安装步骤
购买GPU机器后(有关说明,请参见 )
运行first.sh-这将为后续步骤安装基本实用程序
运行second.sh-按照显示的链接中的说明获取Nvidia的驱动程序。 此存储库中提供了针对Ubuntu 16.04的安装(在second.sh中-默认情况下已注释)
使用nvidia_smi确认正确的安装
运行third.sh-这将安装anaconda,pytorch。
执行多GPU测试
conda激活伯特
python multi_gpu.py
请注意,这30个输入的批次分布在8个GPU上-7个G
2022-04-16 17:56:38
623KB
Shell
1