pytorch2.3 with ROCM6.0

Signed-off-by: Alexander Piskun <[email protected]>
cloud-py-api · Apr 25, 2024 · 6108432 · 6108432
1 parent 01eb71f
commit 6108432
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/Makefile b/Makefile
@@ -25,8 +25,8 @@ build-push:
 	docker login ghcr.io
 	docker buildx build --push --platform linux/arm64/v8,linux/amd64 --tag ghcr.io/cloud-py-api/ai_image_generator_bot:2.1.0 .
 
-.PHONY: build-push-last
-build-push-last:
+.PHONY: build-push-latest
+build-push-latest:
 	docker login ghcr.io
 	docker buildx build --push --platform linux/arm64/v8,linux/amd64 --tag ghcr.io/cloud-py-api/ai_image_generator_bot:latest .
 

diff --git a/lib/hw_install.py b/lib/hw_install.py
@@ -51,8 +51,7 @@ def hw_install():
     if defined_accelerator == "cpu":
         requirements = "torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu"
     elif defined_accelerator == "rocm":
-        # TO-DO: upcoming PyTorch 2.3 will have ROCM 6.0 by default
-        requirements = "--pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0"
+        requirements = "torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0"
     else:
         requirements = "torch torchvision torchaudio"