Skip to content

Instantly share code, notes, and snippets.

@KushajveerSingh
Created October 7, 2021 03:53
Show Gist options
  • Save KushajveerSingh/700ff28547baa8b677944ca401a3bd48 to your computer and use it in GitHub Desktop.
Save KushajveerSingh/700ff28547baa8b677944ca401a3bd48 to your computer and use it in GitHub Desktop.
RUN pip install numpy && \
conda install pytorch torchvision cudatoolkit=11.1 -c pytorch -c nvidia
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment