User Tools

Site Tools


public:docker

Differences

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

Link to this comparison view

Both sides previous revisionPrevious revision
Next revision
Previous revision
public:docker [2025/02/23 21:12] liangpublic:docker [2025/02/23 23:03] (current) – [Step 4: Work with PyTorch] liang
Line 16: Line 16:
     * # exit     * # 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   * 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 ===== ===== References =====
  
-  * https://github.com/saikhu/Docker-Guide-for-AI-Model-Development-and-Deployment+  * https://github.com/saikhu/Docker-Guide-for-AI-Model-Development-and-Deployment (Chapter 6 is obsolete. See the above Step 2 for the latest information)
  
public/docker.1740312724.txt.gz · Last modified: 2025/02/23 21:12 by liang