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

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

暂无评论

发表评论

相关推荐

yolov5训练数据集划分

yolov5训练数据集划分 按照默认8:1:1划分训练集,测试集,验证集。 txt文件出现在imageset文件夹。 import os import randomtrainval_pe

Pytorch—万字入门SSD物体检测

前言 由于初入物体检测领域,我在学习SSD模型的时候遇到了很多的困难。一部分困难在于相关概念不清楚,专业词汇不知其意,相关文章不知所云;另一部分困难在于网上大部分文章要么只是简要介绍了SS

目标检测-锚框概念和代码实现

前言 经历过图像分类后,进一步的就是更复杂的目标检测了,从这一章开始,将会不断记录图像目标检测中的学习经历,其中大多数思路以及代码来源,来自于李沐的动手学深度学习课程&#x