这是困扰我很久的问题,今天花了一点时间去研究一下为什么我的cuda和cudann安装完成了,却不能使用GPU,原因是我的torch和torchvision是cpu版本的,必须下载GPU版本的才能使torch.cuda.is_available()为True,本文使用将文件下载到文件夹后进行安装。
1、打开Anaconda Prompt(Anaconda3)
2、打开后是以base开头的对话框
使用conda activate opencv 此处opencv是我自己创建的环境名称,此处更换为你自己的环境名
进入你自己需要安装torch的位置,输入python,进入python中
import torch
torch.cuda.is_available()
因为我的电脑解决了这个问题,所以是True,如果您的电脑是False,查看是否是CPU版本的torch.
输入torch.__version__ 如果返回cpu字样ÿ