torch.cuda.is_available()返回false

我用的是windows下的miniconda,cuda版本是10.0,安装的pytorch版本是1.7,怀疑是高版本的pytorch不兼容cuda10.0。成功解决办法:
安装cuda10.2和对应cudnn
删除\miniconda3\pkgs文件夹下cuda、pytorch、torchvision等相关的文件夹
重新创建虚拟环境

conda create -n pytorch_1.7 python=3.6

在虚拟环境中安装pytorch

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch 

版权声明:本文为CSDN博主「qq_43506937」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_43506937/article/details/121835324

qq_43506937

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

MMdetection之necks之FPN

MMdetection之necks之FPN 其横向为 1X1 的卷积,向下为 上采样 Specifically, for ResNets [16] we use the feature activations output by

NMS非极大值抑制

IOU IoU 的全称为交并比(Intersection over Union),IoU 计算方法有两种 1、 “预测的边框” 和 “真实的边框” 的交集和并集的比值。 2、 “预测的边框” 和 “真

弱监督目标检测

弱监督目标检测 弱监督目标检测的解决方案 基于多示例学习的弱监 督目标检测算法 基于类激活图的弱监督目标检测算法 基于注意力机制的弱监督目标检测算法 基于伪标签的弱监督目标检测算法 弱监督目标检测主要面临的问题 定位的精度&#xff