conda安装GPU版pytorch,结果总是cpu版本

这篇具有很好参考价值的文章主要介绍了conda安装GPU版pytorch,结果总是cpu版本。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

一.问题

正常安装某版本pytorch时,采用官网的方法和对应的命令,选择适合的pytorch、cudatoolkit版本

Previous PyTorch Versions | PyTorch

由于要在GPU上运行,但测试发现pytorch装的是cpuonly的版本:

python
import torch
torch.cuda.is_available()

得到的结果是false

再检查安装的pytorch版本,可以看到是cpuonly的版本:

conda安装GPU版pytorch,结果总是cpu版本

 二.解决方案

这里建议直接采用解决三

 解决一:

参考:安装pytorch-gpu时会默认安装cpu版本_勇往无前猪猪侠的博客-CSDN博客

很多方法都说conda uninstall cpuonly过后,再重新装就好,但我采用了这个命令后仍然安装的是cpuonly的版本。

一个小tip:删除cpuonly后,最好清空一下conda的包的缓存,否则笨蛋conda看到缓存中包含了要求的pytorch版本,并不会从官网重新下载新的pytorch,装回来的还是cpuonly的版本。指令:

conda clean -p      //删除缓存
conda clean --all 

大部分人遇到的问题在这里已经解决了,尽管如此,我的cpuonly问题仍然没有解决。

解决二:

看到一篇关于分析cpuonly问题的根本原因:https://windses.blog.csdn.net/article/details/125910538r​​​​​​​r

如果你采用的不是默认源,而添加了诸如清华源等其他源,这些源中可能不包含你所要求的cudatoolkit,torchversion,pytorch版本,conda可能确实会自作主张Install别的pytorch了,解决方案参考链接。

解决三:

非常不幸,我关掉了清华源采用默认源,认真卸载了cpuonly和旧的pytorch,也清理了conda缓存包,还按照官网的conda forge指令下载,但conda还是有他自己的想法硬装CPU版。

问题的本质是conda根据你的指令不能选中唯一的包,总是选中其他的安装包,那我们不妨强行锁定包的版本,或者自己下载好wheel在本地pip安装。

举例:我的显卡是3090,必须采用11.0以上的cuda版本,python版本3.8,pytorch因为环境适配原因必须选择1.8.1,进入pytorch相关版本的网站查看所有whl:

https://download.pytorch.org/whl/torch_stable.html

在这里找到我需要的whl:cu111,torch1.8,cp38,linux——点击下载

conda安装GPU版pytorch,结果总是cpu版本

 下载后,进入下载的位置,pip+包名即可安装。

或者不用下载到本地,采用pip指令,比如我采用的是:

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

注意需要在上述网站中找到网站中有的torch\cuda\python兼容的包,没有的会报错。安装其他版本的家人们在相关位置做替换即可。

如果cudatoolkit原本不是这个包的版本,就conda install cudatoolkit=版本,重新安装cudatoolkit即可,只要与自己的GPU适配即可(可在NVIDIA官网查到适配版本)

这样就成功装成了gpu版本的pytorch:

import torch
torch.cuda.is_available()

显示为true

问题解决文章来源地址https://www.toymoban.com/news/detail-462135.html

到了这里,关于conda安装GPU版pytorch,结果总是cpu版本的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 配置(GPU/CPU)Pytorch环境以及修改conda和pip默认安装路径,安装环境前必看!!(———记自己装环境踩过的坑......)

    版本: pytorch2.0 pycharm2022.3 python3.8 如果你要配置pytorch,你先要确定自己在这台电脑上是只作为学习调试代码用还是要用来跑模型,如果用来跑模型,请先下载CUDA和Cudnn,详见此链接安装配置CUDA与cuDNN安装教程(超详细) 如果是单纯用来学习调试代码,不需要调用GPU版本的p

    2024年02月07日
    浏览(60)
  • 在anaconda下安装pytorch + python3.8+GPU/CPU版本 详细教程

    没安装Anaconda的同学可以参考以下安装链接: https://blog.csdn.net/qq_45281807/article/details/112442577 按照安装CPU版本和GPU两个版本进行分类,一般运行程序建议使用CPU版本的,安装更方便。 注意!如果切换镜像后当出现下载不了的情况,就先切换默认源,然后再修改另一个可以使用的

    2024年01月19日
    浏览(74)
  • 怎么把pytorch从CPU版本替换成GPU版本

    使用pip命令 pip uninstall torch 就可以卸载当前的torch版本。(不是cpu版本一般也没有必要重装吧?) 接着找到官网https://pytorch.org/get-started/locally/ 在里面选择 根据你自己的需要选择最新的(我现在是11.8)或者没那么新的(11.7)的GPU版torch 接着复制下面的pip3的安装命令 pip3 ins

    2024年02月13日
    浏览(38)
  • pytorch实现简单的线性回归cpu版本和gpu版本

    实现步骤 准备数据 定义模型 实例化模型,实例优化器类,实例loss 循环进行梯度下降,参数更新 cpu版本实现

    2024年02月16日
    浏览(68)
  • conda 安装pytorch(gpu)

    前提: 确定你安装好了anaconda并配置好了环境变量 ,如果没有,网上有很多详细的配置教程,请自行学习 打开Anaconda Prompt 创建conda虚拟环境并激活 注:env_name表示创建虚拟环境的名字,可以起任意名(我创建时命名为pytorch) 激活虚拟环境 总结: 系统的CUDA版本决定了系统最

    2024年02月02日
    浏览(63)
  • GPU版本pytorch的安装,配套环境python、Cuda、Anaconda安装和版本选择,及常见问题调用gpu返回false

    前言 :第一次装这个我也很懵,就想记录一下交流经验,这个安装最麻烦的是需要各个 版本 都需要 对应 。我也看了很多教程网上基本上安装都是cpu版本,就官网链接安装下来也是cpu版本,然后就不能调用显卡。 本教程使用python3.9、pytorch1.8.0、Cuda11.2版本、Cudnn8.8.1,这个

    2024年02月03日
    浏览(53)
  • 【安装GPU版本pytorch,torch.cuda.is_available()仍然返回False问题】

    @TOC 检查cuda是否安装,CUDA环境变量是否正确设置,比如linux需要设置在PATH,window下环境变量编辑看看,是否有CUDA 首先查看你环境里的pytorch是否是cuda版本,我这里是conda安装的,使用conda list查看是这样: 圈出来的torch带cuda,那安装的肯定时cuda版本,但是torch.cuda.is_available

    2024年02月10日
    浏览(47)
  • pytorch 安装部署(cpu+gpu) linux+windows

    官网打开可能较慢,耐心等待,可以正常打开。 链接:pytorch官网 2023年12月10日更新,演示版本2.1.1 如果中间有任何报错,参考最后一节的处理 1.pytorch是一个和tensorflow类似的框架 如果需要安装tensorflow,可以参考: tensorflow 1,2 cpu+gpu(windows+linux)安装 2.安装anaconda 主要是为

    2024年02月03日
    浏览(43)
  • 将GPU版本的torch和torchvision下载至本地进行安装,解决torch.cuda.is_available()为False的情况,本文主要解决CPU版本的torch问题。

    这是困扰我很久的问题,今天花了一点时间去研究一下为什么我的cuda和cudann安装完成了,却不能使用GPU,原因是我的torch和torchvision是cpu版本的,必须下载GPU版本的才能使torch.cuda.is_available()为True,本文使用将文件下载到文件夹后进行安装。 1、打开Anaconda Prompt(Anaconda3)  2、打开

    2024年02月10日
    浏览(63)
  • pytorch2.0.1 安装部署(cpu+gpu) linux+windows

    官网打开可能较慢,耐心等待,可以正常打开。 链接:pytorch官网 2023年12月10日更新,演示版本2.1.1 如果中间有任何报错,参考最后一节的处理 1.pytorch是一个和tensorflow类似的框架 如果需要安装tensorflow,可以参考: tensorflow 1,2 cpu+gpu(windows+linux)安装 2.安装anaconda 主要是为

    2024年02月09日
    浏览(47)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包