But why does it install old pytorch=1.0.0 version not something >1.6?
If I try to force install pytorch=1.6, it gives the following error:
UnsatisfiableError: The following specifications were found to be incompatible with a past
explicit spec that is not an explicit spec in this operation (cudatoolkit):
- cudatoolkit=11.2
- pytorch=1.6 -> cudatoolkit[version='>=10.1,<10.2|>=10.2,<10.3|>=9.2,<9.3']
The following specifications were found to be incompatible with each other:
Output in format: Requested package -> Available versions
Package setuptools conflicts for:
setuptools
python=3.7 -> pip -> setuptools
conda[version='>=4.10.3'] -> setuptools[version='>=31.0.1']
wheel -> setuptools
pip -> setuptools
...
I basically want to work with the latest versions of pytorch with cudatoolkit=11.2.
No, you cannot install the PyTorch binaries built with CUDA11.2, since they are not available.
If you want to use this specific CUDA version, you could build from source.
Alternatively, you could install the nightly binaries with CUDA11.3 using:
The binaries ship with their own CUDA runtime so your local CUDA toolkit (assuming that’s what you mean by “cuda 11.2”) won’t be used unless you are building from source or a custom CUDA extension.
I’m unsure which “solution” you are referring to, but in case it’s my initial post mentioning to use the supported install commands, then yes. The PyTorch binaries use their own CUDA runtime and your local CUDA toolkit will only be used if you are building from source or a custom CUDA extension.