记录安装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是否可用,成功了!!!!!!!!!