pytorch2.3 with ROCM6.0

Signed-off-by: Alexander Piskun <[email protected]>

Author: bigcat88

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 .
 

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"