change cuda version

Author: falconlee236

README.md

@@ -1,23 +1,26 @@
 # ComputeEngineGPU-Terraform
 
 ## 사용방법
-1. `src/s3_init` 디렉토리로 이동한다.
+### 처음 사용하는 경우
 ```bash
-cd src/s3_init
+make init
 ```
-2. tfstate를 원격으로 관리할 s3와 dynamoDB를 생성한다.
+### 나중에 사용하는 경우
 ```bash
-terraform init
-terraform plan
-terraform apply
+make
 ```
-3. `src` 디렉토리로 이동한다.
+### gcp 인스턴스만 삭제하고 싶은 경우
 ```bash
-cd ..
+make clean
 ```
-4. instance를 생성한다. -> 1, 3시 테스트 성공, 4시 테스트 성공
+
+### aws s3까지 삭제하고 싶은 경우 (완전 초기화)
+```bash
+make fclean
+```
+
+### 꿀팁
+사용가능한 딥러닝 이미지를 보고 싶은 경우
 ```bash
-terraform init
-terraform plan
-terraform apply
+gcloud compute images list --project deeplearning-platform-release | grep cu123
 ```

src/modules/worker/main.tf

@@ -11,8 +11,8 @@ resource "google_compute_instance" "gpu_instance" {
 
     boot_disk {
         initialize_params {
-            // cuda image that works with cuda 11.8
-            image = "deeplearning-platform-release/pytorch-latest-cu118"
+            // cuda image that works with cuda 12.3
+            image = "deeplearning-platform-release/common-cu123-notebooks-ubuntu-2204"
             type = "pd-ssd"
             size = 150
         }
@@ -103,7 +103,11 @@ resource "google_compute_instance" "gpu_instance" {
             "sudo systemctl start docker",
             "sudo systemctl enable docker",
             "echo ${var.dockerhub_pwd} | docker login -u ${var.dockerhub_id} --password-stdin", # dockerhub login
-            "docker pull falconlee236/rl-image:graph-mamba" # pull train docker iamge
+            "docker pull falconlee236/rl-image:parco-cuda123", # pull train docker iamge
+            "sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-key C99B11DEB97541F0", 
+            "sudo apt-add-repository https://cli.github.com/packages", 
+            "sudo apt update", 
+            "sudo apt install gh", # github cli install
          ]
          connection {
             type = "ssh"
@@ -113,15 +117,4 @@ resource "google_compute_instance" "gpu_instance" {
             host = google_compute_instance.gpu_instance.network_interface[0].access_config[0].nat_ip
          }
     }
-}
-
-/*
-단계
-1. 121 cuda 버전 성공 (o)
-2. github action destory 성공 (x)
-3. gcp cloud storage bucket 생성 + 마운트 성공 (o)
-4. docker image container volume 수정하는거 성공
-5. dockerimage 12.1 버전으로 빌드, push + tag
-6. dockerimage 11.7 버전으로 빌드, push + tag
-7. github action apply 성공
-*/
+}