@ashwitha_d Actually, it is a bit tricky for you to switch from CPU build to CUDA build. conda install -c pytorch cudatoolkit=10.2 pytorch torchvision is not enough. You’ll also need to remove cpuonly by doing conda uninstall cpuonly.
@ashwitha_d Actually, it is a bit tricky for you to switch from CPU build to CUDA build. conda install -c pytorch cudatoolkit=10.2 pytorch torchvision is not enough. You’ll also need to remove cpuonly by doing conda uninstall cpuonly.