CUDA Toolkit 9.0 code example
Example: cudatoolkit 11.1 and cudatoolkit 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia -c conda-forge
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia -c conda-forge