Skip to content

Commit 0a2e94d

Browse files
WBobbyyanyao-wang
andauthored
Add ROCm5.2.3/AMDGPU support for PyTorch 1.10 (#1085)
Co-authored-by: Wang, Yanyao <[email protected]>
1 parent 778a37d commit 0a2e94d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.circleci/docker/common/install_rocm.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ ver() {
3030
}
3131

3232
# Map ROCm version to AMDGPU version
33-
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.0.1"]="21.50.1" ["5.1"]="22.10" ["5.1.1"]="22.10.1" ["5.1.3"]="22.10.3" ["5.2"]="22.20" )
33+
declare -A AMDGPU_VERSIONS=( ["4.5.2"]="21.40.2" ["5.0"]="21.50" ["5.0.1"]="21.50.1" ["5.1"]="22.10" ["5.1.1"]="22.10.1" ["5.1.3"]="22.10.3" ["5.2"]="22.20" ["5.2.3"]="22.20.3" )
3434

3535
install_ubuntu() {
3636
apt-get update

0 commit comments

Comments
 (0)