第一个打印语句将显示你安装的PyTorch的版本。第二个打印语句torch.cuda.is_available()将显示一个布尔值,如果PyTorch支持并可以使用GPU,它将返回True,否则返回False。 方法二:查看PyTorch的安装包 如果你知道你的PyTorch是通过某个包管理器(如pip)安装的,你可以检查安装的历史记录来查看是否安装了GPU版本的PyTorch。...
总结一下,要查看Anaconda中安装的PyTorch是否为带GPU版本的,我们可以先使用conda list | grep torch命令确认是否安装了PyTorch,然后使用torch.cuda.is_available()函数判断是否支持GPU计算,最后使用torch.cuda.get_device_name()函数获取当前GPU的名称。 下面是一个使用mermaid语法的关系图,展示了上述过程的流程: erDiag...
CUDA Toolkit (Pytorch)与CUDA Toolkit (nvidia)版本 在安装CUDA Toolkit (Pytorch)时,会下载相应的toolkit包(Pytorch版本与对应CUDA toolkit版本对应关系可以查看官方列表),也就是说,官方发布了自己提前根据对应CUDA toolkit版本编译的pytorch包,编译好的pytorch包也会直接调用下载的轻量版本的CUDA Toolkit (Pytorch)。