项目作者: works-on-my-machine
项目描述 :
Template repository for PyTorch, OpenCV and NLOpt development in Visual Studio Code. Includes CUDA support.
高级语言: Shell
项目地址: git://github.com/works-on-my-machine/pytorch-opencv-vscode.git
pytorch-opencv-vscode
Requirements
- CUDA capable GPU
- nvidia-docker runtime
Features
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
Usage
- Click
Use this template
to create your repository. - Open this folder in VSCode.