【新方法】通過清華鏡像源加速 PyTorch GPU 2.5安裝及 CUDA 版本選擇指南

在這里插入圖片描述

下面詳細介紹所提到的兩條命令,它們的作用及如何在你的 Python 環境中加速 PyTorch 等庫的安裝。

1. 設置清華鏡像源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

這條命令的作用是將 pip (Python 的包管理工具)默認的安裝源更改為 清華大學的鏡像源 ,從而加速后續包的下載速度。清華鏡像源是國內非常可靠的 PyPI 鏡像源之一,尤其是在中國大陸使用時,它能顯著提高安裝 Python 庫的速度。

背景知識:
  • 默認情況下,pip 使用的是 官方 PyPI (Python Package Index)的源,該源位于國外。在中國大陸,由于網絡訪問的限制和延遲,訪問官方 PyPI 源時可能會遇到下載慢、超時等問題。
  • 為了解決這個問題,很多高校和公司提供了 PyPI 鏡像源 ,清華大學提供的鏡像源就其中之一。使用鏡像源可以大大加速下載過程。
如何工作:
  • 執行 pip config set global.index-url 后,pip 會將下載源更改為清華鏡像源。因此,后續所有通過 pip install 安裝的包都會從該鏡像源獲取,而不是默認的官方源。
示例:

假設你需要安裝一些庫,如 NumPyPandas ,在清華鏡像源下下載會比從官方源快得多。你只需要運行:
pip install numpy

該命令會自動從清華鏡像源下載和安裝 numpy 庫。

2. 通過指定 PyTorch 官方源安裝 PyTorch、TorchVision 和 Torchaudio

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

這條命令的目的是通過指定 PyTorch 官方源 下載并安裝 PyTorchTorchVisionTorchaudio 庫。你還指定了 PyTorch 適用于 CUDA 11.8 版本的安裝包。

背景知識:
  • PyTorch 是一個廣泛使用的深度學習框架,支持 GPU 加速。為了使用 PyTorch 的 GPU 功能,你需要安裝支持特定 CUDA 版本的 PyTorch。
  • CUDA 是 NVIDIA 提供的并行計算平臺,支持在 GPU 上進行高效計算。PyTorch 需要與 CUDA 版本匹配才能在 GPU 上運行。
  • 在 PyTorch 的安裝過程中,你可以選擇對應你的 CUDA 版本的安裝包。例如,你指定了 cu118,表示安裝與 CUDA 11.8 兼容的版本。
說明:
  • torch 是 PyTorch 庫的核心,提供了深度學習的基礎功能。
  • torchvision 是一個計算機視覺庫,包含了常見的圖像處理、模型、數據集等功能。
  • torchaudio 是一個用于音頻處理的庫,擴展了 PyTorch 在音頻領域的能力。

你指定的 --index-url https://download.pytorch.org/whl/cu118 是 PyTorch 官方的二進制文件源,它為不同的 CUDA 版本提供了預編譯的 PyTorch 安裝包。這樣可以避免你手動編譯 PyTorch,從而加速安裝過程。

示例:

如果你已經設置了清華鏡像源,并希望安裝適用于 CUDA 11.8 的 PyTorch,可以運行以下命令:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

這會直接從 PyTorch 的官方源下載并安裝支持 CUDA 11.8 的版本,確保你能夠利用 GPU 來加速深度學習訓練和推理。

詳細流程總結:

  1. 設置鏡像源
    • 通過 pip config set global.index-url 命令,將 pip 默認的包源更改為清華鏡像源,能夠有效加速包的下載速度,尤其是對于在中國大陸的用戶。
    • 使用清華鏡像源的好處是,它提供了本地化的源,減少了跨境訪問的延遲。
  2. 安裝 PyTorch 等庫
    • 使用 pip install 命令安裝指定版本的 PyTorch 相關庫,并通過 --index-url 指定官方源(如 CUDA 11.8 版本的 PyTorch)。
    • 通過這種方式,你可以確保安裝的是兼容你機器的 CUDA 版本的 PyTorch,從而能夠利用 GPU 提升計算性能。

額外的注意事項:

  • CUDA 版本 :確保你的機器上已安裝相應版本的 CUDA ,否則無法利用 GPU 加速。不同的 CUDA 版本可能需要不同的 PyTorch 安裝包。
  • 虛擬環境 :建議在 Python 虛擬環境中進行安裝,以避免與系統環境發生沖突。

小貼士:

如果你需要使用不同版本的 CUDA,可以訪問 PyTorch 官方安裝指南,選擇合適的版本并生成對應的安裝命令:PyTorch 官網安裝頁面

參考

https://pytorch.org/get-started

本文來自互聯網用戶投稿,該文觀點僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務,不擁有所有權,不承擔相關法律責任。
如若轉載,請注明出處:http://www.pswp.cn/web/64334.shtml
繁體地址,請注明出處:http://hk.pswp.cn/web/64334.shtml
英文地址,請注明出處:http://en.pswp.cn/web/64334.shtml

如若內容造成侵權/違法違規/事實不符,請聯系多彩編程網進行投訴反饋email:809451989@qq.com,一經查實,立即刪除!

相關文章

【數據結構】單鏈表的使用

單鏈表的使用 1、基本概念2、鏈表的分類3、鏈表的基本操作a、單鏈表節點設計b、單鏈表初始化c、單鏈表增刪節點**節點頭插:****節點尾插:****新節點插入指定節點后:**節點刪除: d、單鏈表修改節點e、單鏈表遍歷,并打印…

虛幻引擎是什么?

Unreal Engine,是一款由Epic Games開發的游戲引擎。該引擎主要是為了開發第一人稱射擊游戲而設計,但現在已經被成功地應用于開發模擬游戲、恐怖游戲、角色扮演游戲等多種不同類型的游戲。虛幻引擎除了被用于開發游戲,現在也用于電影的虛擬制片…

Linux(Centos 7.6)yum源配置

yum是rpm包的管理工具,可以自動安裝、升級、刪除軟件包的功能,可以自動解決軟件包之間的依賴關系,使得用戶更方便軟件包的管理。要使用yum必須要進行配置,個人將其分為三類,本地yum源、局域網yum源、第三方yum源&#…

Linux上更新jar包里的某個class文件

目標:替換voice-1.0.jar里的TrackHandler.class文件 一.查詢jar包里TrackHandler.class所在的路徑 jar -tvf voice-1.0.jar |grep TrackHandler 二.解壓出TrackHandler.class文件 jar -xvf voice-1.0.jar BOOT-INF/classes/com/yf/rj/handler/TrackHandler.cla…

機器學習中回歸預測模型中常用四個評價指標MBE、MAE、RMSE、R2解釋

在機器學習中,評估模型性能時常用的四個指標包括平均絕對誤差(Mean Absolute Error, MAE)、均方誤差(Mean Squared Error, MSE)、均方根誤差(Root Mean Squared Error, RMSE)和決定系數&#xf…

基于SpringBoot的Jwt認證以及密碼aes加密解密技術

目錄 前言 1.SpringBoot項目的創建 2.相關技術 3.項目架構 4.項目關鍵代碼 5.項目最終的運行效果 ?編輯 6.PostMan測試接口結果 前言 學習了SpringBoot之后,才覺得SpringBoot真的很方便,相比傳統的SSH,SSM,SpringBo…

uniapp下載打開實現方案,支持安卓ios和h5,下載文件到指定目錄,安卓文件管理內可查看到

uniapp下載&打開實現方案,支持安卓ios和h5 Android: 1、申請本地存儲讀寫權限 2、創建文件夾(文件夾不存在即創建) 3、下載文件 ios: 1、下載文件 2、保存到本地,需要打開文件點擊儲存 使用方法&…

77、將adaface的mtcnn模型npy文件轉成atlas310p模型,并進行推理

基本思想:將adaface的mtcnn模型npy文件轉成atlas310p模型進行推理。同時比對結果 ubuntu@ubuntu:~$ git clone https://github.com/mk-minchul/AdaFace.git Cloning into AdaFace... remote: Enumerating objects: 236, done. remote: Counting objects: 100% (109/109), don…

Spark SQL DML語句

【圖書介紹】《Spark SQL大數據分析快速上手》-CSDN博客 《Spark SQL大數據分析快速上手》【摘要 書評 試讀】- 京東圖書 Spark本地模式安裝_spark3.2.2本地模式安裝-CSDN博客 DML(Data Manipulation Language,數據操作語言)操作主要用來對…

農歷節日倒計時:基于Python的公歷與農歷日期轉換及節日查詢小程序

農歷節日倒計時:基于Python的公歷與農歷日期轉換及節日查詢小程序 摘要 又是一年春節即將到來,突然想基于Python編寫一個農歷節日的倒計時小程序。該程序能夠根據用戶輸入的農歷節日名稱,計算出距離該節日還有多少天。通過使用lunardate庫進…

線性直流電流

電阻網絡的等效 等效是指被化簡的電阻網絡與等效電阻具有相同的 u-i 關系 (即端口方程),從而用等效電阻代替電阻網絡之后,不 改變其余部分的電壓和電流。 串聯等效: 并聯等效: 星角變換 若這兩個三端網絡是等效的,從任…

CDN(Content Delivery Network,內容分發網絡)

CDN(Content Delivery Network,內容分發網絡)是一種通過在網絡中部署分布式的服務器集群,將網站內容分發到最接近用戶的服務器節點,以提高用戶訪問速度和穩定性的重要網絡基礎設施。CDN的核心思想是讓用戶就近獲取所需…

B站推薦模型數據流的一致性架構

01 背景 推薦系統的模型,通過學習用戶歷史行為來達到個性化精準推薦的目的,因此模型訓練依賴的樣本數據,需要包括用戶特征、服務端推薦的視頻特征,以及用戶在推薦視頻上是否有一系列的消費行為。 推薦模型數據流,即為…

【LeetCode】839、相似字符串組

【LeetCode】839、相似字符串組 文章目錄 一、并查集1.1 并查集 二、多語言解法 一、并查集 1.1 并查集 求共有幾組, 聯想到并查集, 即并查集有幾個集合 字符串相似: 相差0個字符, 或2個字符 其中所有字符串長度都相同, 是比較方便處理的 // go var sets int var father […

你不需要對其他成年人的情緒負責

在這個紛繁復雜的世界里,每個人都是獨一無二的個體,背負著各自的故事、夢想與煩惱。在人際交往的廣闊舞臺上,我們時常會遇到這樣的情境:朋友、同事、家人,甚至是陌生人,他們的情緒似乎總能不經意間影響到我…

官宣!低空經濟司,掛牌成立!

近日,國家發展改革委網站“機關司局”欄目悄然更新,一個新設立的部門——低空經濟發展司(簡稱“低空司”)正式進入公眾視野。低空司的成立,無疑是對當前國家經濟發展形勢的深刻把握和前瞻布局。 低空經濟是以各類低空飛…

接口調用限頻(代理模式+滑動窗口)

目錄 代碼示例 接口 代理 接口實現 限流工廠 限流處理器接口 直接交換處理器 限流處理器 限流配置 滑動窗口限流 通過代理模式滑動窗口,限流請求第三方平臺,避免出現第三方平臺拋出限流異常,影響正常業務流程,從出口出發…

不安全物聯網的輕量級加密:綜述

Abstract 本文綜述了針對物聯網(IoT)的輕量級加密解決方案。這項綜述全面覆蓋了從輕量級加密方案到不同類型分組密碼的比較等多個方面。同時,還對硬件與軟件解決方案之間的比較進行了討論,并分析了當前最受信賴且研究最深入的分組…

【小程序】全局數據共享

目錄 全局數據共享 1. 什么是全局數據共享 2. 小程序中的全局數據共享方案 全局數據共享 - MobX 1. 安裝 MobX 相關的包 2. 創建 MobX 的 Store 實例 3. 將 Store 中的成員綁定到頁面中 4. 在頁面上使用 Store 中的成員 ?5. 將 Store 中的成員綁定到組件中 6. 在組件中…

自動化測試- 自動化測試模型

目錄 自動化測試模型簡介 1、線性模型 舉例 測試頁面html文件 測試腳本 2. 關鍵字驅動測試(Keyword-Driven Testing) 需測試內容 關鍵字驅動測試框架 創建測試用例文件 運行測試 3. 數據驅動測試(Data-Driven Testing) …