首页 > 其他分享 >安装GPU版本Pytorch失败:torch.cuda.is_available()为False

安装GPU版本Pytorch失败:torch.cuda.is_available()为False

时间:2024-01-17 22:23:42浏览次数:30  
标签:available False torch Pytorch conda cuda 版本 GPU 安装

问题所在

检查conda list发现,实际安装的Pytorch为CPU版本(虽然安装时明确指定了cuda版本):
image
上图中可以看出,Pytorch的描述为:py3.9_cpu_0

解决办法

有可能是因为环境中存在一个叫“cpuonly”的包,导致无法安装GPU版本Pytorch:
image
卸载掉它即可,卸载它时会自动将Pytorch更新为GPU版本:
image
如果检查发现环境中其实并没有这个包呢?可以借鉴此方法,先安装一个“cpuonly”包,再卸载掉它,Pytorch也会自动更新为GPU版本。

conda unistall pytorch
conda istall pytorch

标签:available,False,torch,Pytorch,conda,cuda,版本,GPU,安装
From: https://www.cnblogs.com/yutian-blogs/p/17971315

相关文章

  • 查看Stable Diffusion pytorch版本
    查看StableDiffusionpytorch版本简介StableDiffusion是一种用于生成模型的无监督学习算法,它可以通过不断扩散噪声来生成逼真的样本。在这篇文章中,我们将介绍如何查看StableDiffusion的pytorch版本,并提供一些代码示例。安装首先,我们需要安装pytorch和torchvision......
  • 在 SAP Web IDE 个人版中根据模板创建项目时,选择 OData 服务时出现catalog service is
     1.NOTE2684697 ,重点是点2点5的问题 2.去掉CatalogServiceVersion2的系统别名(包括LOCAL)翻阅其他博客,有人说是因为系统别名,我给去掉了。 ......
  • Kubernetes集群中 Pod 中报pthread_create failed: Resource temporarily unavailable
    查看containerd日志显示如下信息OCIruntimeexecfailed:execfailed:unabletostartcontainerprocess:readinit-p:connectionresetbypeer:unknown"查看Pod日志中显示pthread_createfailed:Resourcetemporarilyunavailable通过监控系统查看节点的进程数t......
  • stable diffusion pytorch cuda 版本
    StableDiffusion:APyTorchCUDAVersionIntroductionDeeplearninghasrevolutionizedvariousfields,includingcomputervision,naturallanguageprocessing,androbotics.PyTorch,apopulardeeplearningframework,hasgainedsignificantattentionduet......
  • 使用pytorch加载llama
    使用PyTorch加载LLAMA数据集在深度学习中,数据集的选择和处理对于模型的性能和训练效果起着至关重要的作用。PyTorch是一个常用的深度学习框架,它提供了各种工具和函数来加载和处理各种常见的数据集。在本文中,我们将介绍如何使用PyTorch加载LLAMA数据集,并提供相应的代码示......
  • 安装torch2trt中遇到的报错:ModuleNotFoundError: No module named 'torch2trt.flatten
    这一个报错折腾了一整天,在这里记录一下方便他人可以快速解决问题:首先说明一下本人的版本情况:Linux系统CUDA:11.8Python:3.8torch:2.0.0torchvision:0.15.0代码中报错的位置是:fromtorch2trtimporttorch2trt报错的原因是:ModuleNotFoundError:Nomodulenamed'torch2trt......
  • PyTorch深度学习零基础入门
    在人工智能时代,机器学习技术日新月异,深度学习是机器学习领域中一个全新的研究方向和应用热点,它是机器学习的一种,也是实现人工智能的必由之路。深度学习的出现不仅推动了机器学习的发展,还促进了人工智能技术的革新。PyTorch作为深度学习的重要框架,近年来备受读者喜爱,自推出后得到了......
  • pytorch调用训练的模型
      importtorch#加载模型参数model_path='model.pth'model_state_dict=torch.load(model_path)#创建模型实例input_size=10#输入层大小,根据实际情况调整hidden_size1=32#第一层隐藏层大小,根据实际情况调整hidden_size2=16#第二层隐藏层大小,根......
  • 深度学习pytorch常用操作以及流程
    在微信公众号上看到这篇文章,担心以后想找的时候迷路,所以记录到了自己的博客上,侵扰致歉,随时联系可删除。1.基本张量操作1.1创建张量介绍:torch.tensor()是PyTorch中用于创建张量的基本函数。简单使用:importtorch#创建一个标量(零维张量)scalar_tensor=torch.tensor(......
  • PyTorch中的nn.LeakyReLU()、nn.Module和nn.ModuleList
    一.nn.LeakyReLU()函数  在PyTorch中,nn.LeakyReLU()是一个激活函数,用于引入非线性性到神经网络中。LeakyReLU是修正线性单元(ReLU)的一种变体,它在输入为负数时不是完全置零,而是引入一个小的负斜率。nn.LeakyReLU()的初始化参数如下:negative_slope(默认为0.01):负斜率,指定当......