I am shifting to using PyTorch from Keras and TensorFlow. For both of those, the setup on Anaconda is fairly simple.
conda install keras-gpu

One command does quick work of installing all libraries including cudatoolkit and keras recognizes my GPU.
I installed pytorch and tried running Chatbot example by pytorch on my GPU (GTX 1050 ti) but it doesn’t seem to recognize my device.
Can anyone illustrate the process I need to follow for this?
Do i need to install CudaToolKit separately?
Thank you.