- Delete dockerfile and docker compose
- copy dockerfile and compose in either mac image for testing on a non gpu machine, or gpu image to train on a gpu machine
- only if using gpu download cudnn 8.4.x and place it in extras/lib
- run docker compose up -d or docker build to make the image
- for gpu, you'll need to install nvidia drivers and nvidia-docker on your machine
- Be aware that this is very shallow detail, there maybe errors or edit that you'll run into that i cannot think about at the moment, for e.g, your cudnn download in extras/lib, will have to be written into extras/shell/cuda.sh, so that it can be installed. Other similar issues may exist. Be ware.
mo379 / tensornets-gpu-jax-docker Goto Github PK
View Code? Open in Web Editor NEWTensornetwork hosted inside a docker container that is gpu enabled running jax