文件名称：multi_gpu_test:在多GPU机器上使用PyTorch进行并行化
文件大小：623KB
文件格式：ZIP
更新时间：2024-05-13 05:55:19
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)