Repo that allows me to build AI tools on top of Hugging Face
numba -s | grep cuda
python utils/verify_cuda_pytorch.py
nvidia-smi
should show a GPU用一个python3环境即可,安装aconda,jupyter notebook,tensorflow,pytorch等, VSCode安装相应的扩展,Nvidia
ai\.devcontainer\Dockerfile
FROM mcr.microsoft.com/vscode/devcontainers/python:0-${VARIANT}
```
镜像一般都比较大
环境配置
环境一般通过 requirements.txt(pip), pyproject.toml (poetry), environment.yml(conda) 管理依赖。