404 B
404 B
Setup
- Install Docker on your local system
- Build docker image and run the shell
- Get your API key from https://wandb.ai/settings#api, docker will automatically connect to WanDB.
docker build -t gpu api_key="<wandb_api_key>" .
docker run --rm -it --gpus all --entrypoint /bin/bash gpu
- To double check if tensorflow is configured properly run
python3 gpu_check.py
.