解决torch.cuda.is_available()返回False的问题(使用pip成功安装)

记录安装pytorch遇见的坑,主要是解决torch.cuda.is_available()返回False的问题。

问题如下
在这里插入图片描述
此问题过于玄学,尝试了各种各样的解决办法,依然没有能够解决,这里记录一下,网上的方法基本都是利用conda进行下载,此处通过测试都失败了,因此采用pip进行下载,下载之后测试成功。

1.电脑版本

GPU:GTX050Ti
CUDA Version:12.0(之前是11.6,不知道为什么自动升级为12.0了,也可使用此命令下载cuda)
Driver Version:526.47

在这里插入图片描述

2.网络查询

2.1可能原因

1.下载了cpu版本
2.版本不对

3.解决方案

3.1 方法1 采用pip安装

可能是conda自动下载了cpu版本,因此采用pip安装

pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116

创建了一个名叫pytorch_pip的环境,尝试此代码
在这里插入图片描述
下载后进行检查gpu是否可用,成功了!!!!!!!!!

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_43471945/article/details/127703461