I created a fresh conda environment and ran the following code,
conda install pytorch torchvision torchaudio cudatoolkit=10.0 -c pytorch on ubuntu 16.04 LTS. Why did this install pytorch version 1.4.0 and not the latest pytorch 1.7.1?
I want to install the latest version.
I am not sure if pytorch is available with cuda 10.0.
Can you try with
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch as given in https://pytorch.org/?
Yes I tried that immediately after and it was installing the latest one. So may be I have to update cuda drivers. Thankyou
The CUDA drivers are needed to run the CUDA runtime, which ships with the PyTorch binaries. For the latest PyTorch version you can install the CUDA runtime in version 9.2, 10.1, 10.2, and 11.0 as @InnovArul described.