Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bump to Cuda 11.1; Implement fix for #157 #174

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

stellarpower
Copy link

Main motivation for this was to support newer cards - we couldn't run this on our hardware with the older version of Cuda; resultantly, pytorch and etc. have also been updated. Tried to follow some guides online as to the correct interdependent versions of each.

As part to this I've made some minimal changes to the Dockerfile to use more variables and tidy; if we continue using this project, would happily provide some more as I believe it could be improved and made a bit easier and more decoupled.

We also ran into bug #157 again so I've pulled in the fix for this from the comments and seems to work for the minimal example training.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant