Skip to content

Latest commit

 

History

History
14 lines (10 loc) · 296 Bytes

NOTES.md

File metadata and controls

14 lines (10 loc) · 296 Bytes

Notes

miscellaneous notes.

How to install pytorch on conda with gpu

mamba install pytorch=1.12 torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

How to make the docker image using the Dockerfile

cargo clean && docker build --progress=plain -f Dockerfile .