Nvidia driver installed:
- Driver: 511.79
- Cuda: 11.5
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch-nightly
- torch==1.12.0.dev20220319
- torchaudio==0.12.0.dev20220319
- torchvision==0.13.0.dev20220319
Nvidia driver installed:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch-nightly