Hello,
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?
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.