- CUDA capable GPU
- nvidia-docker runtime
Template repository that contains:
torch==1.8.1+cu111
torchvision==0.9.1+cu111
torchaudio==0.8.1
opencv-python==4.5.1.48
nlopt==2.7.0
- CUDA 11.1
- Python 3.8 (with intellisense)
- X11 Redirection
- Click
Use this template
to create your repository. - Open this folder in VSCode.