Differences

This shows you the differences between two versions of the page.

--- public:docker [2025/02/23 21:14] – [References] liang
+++ public:docker [2025/02/23 23:03] (current) – [Step 4: Work with PyTorch] liang
@@ Line 16: / Line 16: @@
     * # exit
-===== Step 2 (optional): Install Nvidia GPU support =====
+===== Step 2 (if you have an NVidia GPU): Install Nvidia GPU support =====
   * Follow https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
+===== Step 3: Install PyTorch (for development) =====
+  * $ docker pull pytorch/pytorch:latest
+  * $ docker run --rm --gpus all pytorch/pytorch:latest python -c "import torch; print(torch.cuda.is_available())"
+  * If you see "True", it worked.
+===== Step 4: Work with PyTorch =====
+  * $ docker run -it --gpus all --rm pytorch/pytorch:latest bash
+  * # nvidia-smi
+  * If you see the use of your GPU, it worked.
 ===== References =====