Skip to content

Commit

Permalink
Merge pull request #28 from probcomp/readme-patch-1
Browse files Browse the repository at this point in the history
Readme patch 1
  • Loading branch information
arijitnoobstar authored Oct 23, 2023
2 parents 6517b16 + 58b8acb commit 6b1cd29
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,16 @@ Run:
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
```

Error:
```
raise RuntimeError("Ninja is required to load C++ extensions")
```
Run:
```
sudo apt-get update
sudo apt-get install ninja-build
```

To check your CUDA version:
```
nvcc --version
Expand All @@ -84,7 +94,9 @@ nvcc --version
# GCP Setup
- Start new VM instance (see [link](https://cloud.google.com/compute/docs/instances/create-start-instance)). Select GPU - NVIDIA V100 and Machine Type 8vCPU 4 Core 30GB.

-From the VM instances page, searched for public image `c2-deeplearning-pytorch-2-0-gpu-v20230807-debian-11-py310`. Increase storage to 1000GB.
-From the VM instances page, searched for public image `c2-deeplearning-pytorch-2-0-gpu-v20230925-debian-11-py310`. Increase storage to 1000GB.

- Note that public image names get updated frequently, so it is possible you may not find the one mentioned above. To find the latest public image, go to the [public list](https://cloud.google.com/compute/docs/images#console), and look for an image as close to the one above (Debian 11, CUDA 11.8, Python 3.10, Pytorch 2.0 etc.).

- SSH into instance and when prompted, install the NVIDIA drivers.

Expand Down

0 comments on commit 6b1cd29

Please sign in to comment.