search

uabrc / uabrc.github.io

UAB Research Computing Documentation
https://docs.rc.uab.edu
21 stars 12 forks source link

Add info on PyTorch GPU issues to ask.ci #438

Closed wwarriner closed 1 year ago

wwarriner commented 1 year ago

Waiting on #437 to be merged.

Answer at this link: https://ask.cyberinfrastructure.org/t/why-is-gpu-not-found-after-installing-pytorch-via-anaconda/2621

There are two common reasons a GPU can't be found when using PyTorch.

1. Be sure you've requested a GPU.
    - On Cheaha you will need to use the flags `--partition=pascalnodes` or `--partition=pascalnodes-medium` and `--gres=gpu:1` for a single GPU.
    - With Open OnDemand Jupyter Notebook, you will need to select the `pascalnodes` or `pascalnodes` medium partition. At this time exactly one GPU is automatically requested. It is not possible to request more.
2. Be sure you've installed the GPU flavor of PyTorch. Please see [link] for more information about how to install PyTorch with GPU compatibility.
wwarriner commented 1 year ago

Resolved here: https://ask.cyberinfrastructure.org/t/why-is-gpu-not-found-after-installing-pytorch-via-anaconda/2621