Skip to content

Instantly share code, notes, and snippets.

@maciejskorski

maciejskorski/gist:f0b2170dd7b75d5e9d7a93f1350287de

Created July 25, 2024 13:52
Show Gist options
  • Save maciejskorski/f0b2170dd7b75d5e9d7a93f1350287de to your computer and use it in GitHub Desktop.
Save maciejskorski/f0b2170dd7b75d5e9d7a93f1350287de to your computer and use it in GitHub Desktop.
Download ZIP
nvidia torch container with current dir mounted
Raw
gistfile1.txt
# select an appropriate container tag https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch/tags
docker run -it -v ${PWD}:/workspace --name torch_container --gpus all nvcr.io/nvidia/pytorch:21.07-py3
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment