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.

Sign up for GitHub

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

Jump to bottom

Provide instructions for GPU based pytorch #6

Closed

fire opened this issue Jun 18, 2021 · 1 comment

fire commented Jun 18, 2021 •

edited

Similar to the current cpu version. Provide an environment.yml for cuda based pytorch.

It is non trivial to switch between gpu and cpu. https://github.com/pytorch/pytorch/issues/30664.

The text was updated successfully, but these errors were encountered:

Author

fire commented Jun 18, 2021

The way to do it to create a new environment and run something similar to:

conda install tensorboard tqdm install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

Mamba solves dependencies faster than conda.

fire closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment