I am having problem installing torch and torchvision in conda environment could someone suggest a compatible version for cuda 12.1 driver
Hi, you can try installing cudatoolkit package in the conda enviroment, pytorch should use the one in the environment instead, and conda will make sure the version is compatible with the rest of the packages installed. You can add the nvidia channel as well with -c nvidia so that conda can find more versions.