Help: Installing Pytorch with GTX 1050 Ti

#1

Hello,

I have been using google colab but I have to work on my laptop (not allowed to use google colab).
I couldnt find a definite answer on the version of pytorch to install. I believe is the following, but can someone help please.

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch

or

pip install torch==1.7.0+cu92 torchvision==0.8.0+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

If I am not mistaken, the GTX 1050 Ti support cuda up to 9.2?

Thank you

#2

Hello, there is a great post on the developer.nvidia forum which I have linked for you: GTX1050 TI CUDA. This states, that at least 9.1 is supported.

#3

No, your 1050 Ti is using compute capability 6.1 and is supported in the current PyTorch release (1.13.0) using the 11.6 and 11.7 CUDA runtime, so no need to downgrade to an older release.

#4

This is amazing, thanks a lot.