update

gpu.py

@@ -8,4 +8,4 @@ tensor = torch.rand(10240, 10240).cuda()
 
 while True:
     _ = torch.matmul(tensor, tensor)
-    time.sleep(1)
+

py38.sh

@@ -21,7 +21,7 @@ set -o nounset                              # Treat unset variables as an error
 
 sudo apt update
 
-sudo apt install python3.8-dev python3-pip
+sudo apt install -y python3.8-dev python3-pip
 
 curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 sudo python3.8 get-pip.py

rocm52.py38.sh → rocm52.py38.ds.sh

@@ -1,9 +1,9 @@
 #!/bin/bash - 
 #===============================================================================
 #
-#          FILE: py38.sh
+#          FILE: rocm52.py38.ds.sh
 # 
-#         USAGE: ./py38.sh 
+#         USAGE: ./rocm52.py38.ds.sh 
 # 
 #   DESCRIPTION: 
 # 
@@ -21,7 +21,8 @@ set -o nounset                              # Treat unset variables as an error
 
 sudo apt update
 
-sudo apt install python3.8-dev python3-pip
+sudo apt install -y python3.8-dev python3-pip
+sudo apt install -y ninja-build
 
 curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 sudo python3.8 get-pip.py
@@ -30,3 +31,5 @@ sudo rm -rf get-pip.py
 
 sudo pip3.8 install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/rocm5.2
 
+sudo pip3.8 install deepspeed ninja
+