I am trying to run a pytorch code in gpu. Cuda version 11.7 is installed in my system. Steps i followed to run my file in GPU :
Created conda environment
Installed pytorch using the following command
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=11.0 -c pytorch
(as my code has some dependency,i am using these versions)
Then while i am running the code , it is not using GPU
torch.cuda.is_available() is giving false.
Can anyone help me out with this? what else is required to successfully run my file in gpu.