Repo that allows me to build AI tools on top of Hugging Face
numba -s | grep cuda
python utils/verify_cuda_pytorch.py
run nvidia-smi
should show a GPU
conda config --set solver libmamba
conda create -n LLMs python=3.10
conda activate dl-workshop
conda install jupyterlab watermark
pip install torch==2.0.1
pip install -r requirements.txt
用一个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) 管理依赖。