public:docker
Differences
This shows you the differences between two versions of the page.
Both sides previous revisionPrevious revisionNext revision | Previous revision | ||
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 | * # exit | ||
- | ===== Step 2 (optional): Install Nvidia GPU support ===== | + | ===== Step 2 (if you have an NVidia GPU): Install Nvidia GPU support ===== |
* Follow https:// | * Follow https:// | ||
+ | ===== Step 3: Install PyTorch (for development) ===== | ||
+ | |||
+ | * $ docker pull pytorch/ | ||
+ | * $ docker run --rm --gpus all pytorch/ | ||
+ | * If you see " | ||
+ | |||
+ | ===== Step 4: Work with PyTorch ===== | ||
+ | |||
+ | * $ docker run -it --gpus all --rm pytorch/ | ||
+ | * # nvidia-smi | ||
+ | * If you see the use of your GPU, it worked. | ||
===== References ===== | ===== References ===== |
public/docker.1740312896.txt.gz · Last modified: 2025/02/23 21:14 by liang