奋斗一晚上解决BUG记录一下,因为全网几乎没有遇上我这个bug
这个配置BUG的历程记录: 在英伟达官网下载CUDA,默认推荐给我11.8版本,但是torch最高支持到11.7,在配置11.8的时候,配置了11.7的torch(BUG#1)。 卸载CUDA11.8后,重新下载11.7,再配置torch后,就是如下一直FALSE的输出,很折磨人。 后续大佬查出是因为前面配置的torch有冲突,卸载后,重新下载就好,细节如下。 因为需要使用CUDA和Pytorch
import torch
print(torch.cuda.device_count())
#可用gpu数量
print(torch.cuda.is_available()) # #是否可用gpu
print(torch.version.cuda)
print(torch.__version__)
执行上述代码一直是FALSE和不显示cuda的信息 去某平台求助了大神找到解决方案
- 终端输入
conda activate 你的环境 pip list- 我的终端如下
然后卸载
终端输入
pip uninstall torch torchvision torchaudio 一直按y
后面重新在Pytorch.org 下载对应的版本就ok
注意CUDA和torch版本需对应