We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
Somehow conda env -f ... installs pytorch-cpu on cuda toolkit v10.1.
conda env -f ...
Tested only on Linux-GPU (DSVM NC6v3)
Update cuda from v10.0 to v10.1 and create conda env by following SETUP.md
Either manually update torch after env creation by running:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
or specify cuda version on environment.yaml like:
... dependencies: ... - pytorch=1.3.1=py3.6_cuda10.1.243_cudnn7.6.3_0 - torchvision=0.4.2=py36_cu101 ...
fixes it (i.e. install gpu torch).
The text was updated successfully, but these errors were encountered:
Maybe add this contents to SETUP.md for those who have similar issues.
Sorry, something went wrong.
No branches or pull requests
Description
Somehow
conda env -f ...
installs pytorch-cpu on cuda toolkit v10.1.In which platform does it happen?
Tested only on Linux-GPU (DSVM NC6v3)
How do we replicate the issue?
Update cuda from v10.0 to v10.1 and create conda env by following SETUP.md
Expected behavior (i.e. solution)
Either manually update torch after env creation by running:
or specify cuda version on environment.yaml like:
fixes it (i.e. install gpu torch).
Other Comments
The text was updated successfully, but these errors were encountered: