docker images

[skunkwerk]

Could someone please share a working Docker image for CUDA 12 and PyTorch 2.3?
I've tried building my own, but fails to run properly (the conda version complains there's no nvidia driver, the pip version results in a messed up pytorch).

thanks

[danielhanchen]

Apologies on the delay - just relocated to SF, hence the delay!

If you're using Conda - try copying the Kaggle install instructions here: https://www.kaggle.com/code/danielhanchen/kaggle-llama-3-8b-unsloth-notebook

[dsingal0]

@skunkwerk I have a dockerfile and docker image with torch 2.3 and cuda 12.1
#738