【文件属性】:
文件名称:multi_gpu_test:在多GPU机器上使用PyTorch进行并行化
文件大小:623KB
文件格式:ZIP
更新时间:2021-04-23 12:08:39
Shell
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
【文件预览】:
multi_gpu_test-master
----multi_gpu.png(349KB)
----tf_test.py(581B)
----second.sh(648B)
----ubuntu_16_04_nvidia_driver.sh(580B)
----tensorflow_install.sh(366B)
----first.sh(240B)
----LICENSE(1KB)
----third.sh(2KB)
----multi_gpu.py(1KB)
----README.md(2KB)
----install.png(298KB)