onnxruntime/orttraining/tools/amdgpu
Weixing Zhang dce76c15e7
add dockfile for ROCm 4.2 (#7749)
* add dockfile for ROCm 4.2
* using rocm/pytorch:rocm4.2_ubuntu18.04_py3.6_pytorch_1.8.1
2021-06-08 08:02:27 -07:00
..
model
script Add GPU kernels for ROCm EP (#5655) 2020-11-06 16:11:06 -08:00
Dockerfile.rocm3.7 add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.7.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.8.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.9 add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.9.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.10.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm4.0.pytorch update Dockerfile for workaround for issue in RCCL for rocm4.0 (#7108) 2021-03-23 13:36:04 -07:00
Dockerfile.rocm4.1.pytorch Clean up ROCm4.1 Dockerfile build directory (#7732) 2021-05-20 10:04:49 -07:00
Dockerfile.rocm4.2.pytorch add dockfile for ROCm 4.2 (#7749) 2021-06-08 08:02:27 -07:00