当前位置: 首页 > news >正文

解决下载支持gpu的pytorch问题

是这样的,因为要下载可以用gpu的pytorch,用的pycharm下载,一直尝试

(D-learning) PS D:\pytorch_whl> pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

不限于关防火墙,用conda,翻墙,用清华源,改文件等等,下载是可以下载,但是速度极其慢,网速达到1m/s,但是一在pycharm里面下载就260kb/s,特别慢而且网速不稳定还会中间断开,b溃。

最后的解决办法是,直接在

1. 去 PyTorch 官方 whl 下载页面: 🔗 https://download.pytorch.org/whl/cu121https://download.pytorch.org/whl/cu121/

找到torch,点进去找到你的版本。

(上面是我的版本,如果你想下cuda12.4的或者其他版本改一下后缀)

找到 适合你的 Python 版本和操作系统的 .whl 文件,例如:

  • Windows + Python 3.10

torch-2.4.0+cu121-cp310-cp310-win_amd64.whl(只需要下载这个就可以)
torchvision-0.19.0+cu121-cp310-cp310-win_amd64.whl
torchaudio-2.4.0-cp310-cp310-win_amd64.whl

下载后,将 .whl 文件放入同一个文件夹(比如 D:\pytorch_whl\

这三个 .whl 文件的作用分别是:

  1. torch-2.4.0+cu121-cp310-cp310-win_amd64.whl
    核心 PyTorch 库,必须安装!
  2. torchvision-0.19.0+cu121-cp310-cp310-win_amd64.whl
    用于计算机视觉(CV)任务,比如图像处理、预训练模型等。如果你不做 CV 任务,可以不装。
  3. torchaudio-2.4.0-cp310-cp310-win_amd64.whl
    用于音频处理,比如 ASR(语音识别)等。如果你不做音频任务,可以不装。

2. 手动安装 PyTorch

1️⃣ 打开 CMD 或 PowerShell,或者pycharm终端,切换到 .whl 文件所在目录

cd D:\pytorch_whl\

2️⃣ 安装 PyTorch

pip install torch-2.4.0+cu121-cp310-cp310-win_amd64.whl
pip install torchvision-0.19.0+cu121-cp310-cp310-win_amd64.whl
pip install torchaudio-2.4.0-cp310-cp310-win_amd64.whl

注意

  • cp310 代表 Python 3.10,如果你的 Python 版本不同,请下载对应的 .whl 文件!
  • win_amd64 代表 Windows 64 位,如果你是 Linux,文件名会是 manylinux_x86_64.whl

    3. 安装 PyTorch 依赖

    PyTorch 官方下载的 .whl 文件 中,已经包含了 核心 CUDA 依赖,但有些 Python 包需要手动安装。

    🔹 常见依赖包(可直接用 pip 安装):

    pip install numpy filelock requests packaging pillow tqdm typing-extensions(直接复制粘贴进终端)

    如果你要用 torch-tensorrt(加速推理)

    pip install torch-tensorrt

    如果你要用 torchserve(模型部署)

    pip install torchserve torch-model-archiver

     4. 验证 PyTorch 是否安装成功

    运行

    import torch
    print(torch.__version__)  # 应该输出 2.4.0
    print(torch.cuda.is_available())  # 如果是 True,说明 CUDA 可用

    如果 torch.cuda.is_available() 输出 False

    可能 CUDA 版本不匹配,检查 torch.version.cuda

    print(torch.version.cuda)  # 应该输出 12.1

    确保安装了 NVIDIA 驱动和 CUDA 12.1

    另外我还有些问题想问:

    为什么我的网速很快,都是pycharm下载东西特别慢,不管是否开了镜像都很慢。

    另外为什么我一翻墙想用vpn下载,别人用vpn会更快,但是我就会出现以下这种情况,SSLerror问题。有没有遇到过相同问题的友友,是怎么解决的?

    相关文章:

  1. 云原生网络篇——万级节点服务网格与智能流量治理
  2. [密码学实战]Java生成SM2根证书及用户证书
  3. 【零基础C语言】第四节 数组
  4. 系统架构设计师—计算机基础篇—计算机网络
  5. leetcode 59. 螺旋矩阵 II 中等
  6. DeepSeek效应初现:Grok-3补刀ChatGPT,OpenAI已在ICU?
  7. 如何理解语言模型
  8. 【算法】875. 快速幂
  9. 代码随想录刷题day35|(二叉树篇)二叉树的非递归遍历(前序+后序)
  10. 解决各大浏览器中http地址无权限调用麦克风摄像头问题(包括谷歌,Edge,360,火狐)后续会陆续补充
  11. Mac mini M4安装nvm 和node
  12. 化学工业领域 - 基础化工、精细化工、煤化工极简理解
  13. (十一)基于vue3+mapbox-GL实现模拟高德实时导航轨迹播放
  14. REACT学习第三幕--沉睡花园
  15. Pseudo-Q: Generating Pseudo Language Queries for Visual Grounding
  16. Java类加载机制 双亲委派机制 八股速记版
  17. 网络配置的基本信息
  18. MFC中CMutex类和CSingleLock类,配合使用疑惑
  19. windows电脑上安装llama-factory实现大模型微调
  20. 校园订餐微信小程序(全套)
  21. 怎么在各大网站做推广/seo优化上海牛巨微
  22. 怎么寻找做有益做网站的客户/百度搜索关键词优化
  23. 无极网络是什么意思/湖南seo优化报价
  24. 策划公司电话/seo搜索推广费用多少
  25. html网页制作企业类网站/梧州网站seo
  26. 设计商业网站应该做到什么/网站seo优化排名