【深度學習】pytorch深度學習框架的環境配置

文章目錄

  • 1. 配置cuda環境
  • 2. 配置conda環境
  • 3. 配置pytorch gpu環境

1. 配置cuda環境

  • 在命令行輸入以下命令可以查看當前顯卡驅動版本和最高支持的cuda版本
nvidia-smi

在這里插入圖片描述

  • 根據cuda版本去官網下載并安裝cuda
    下載鏈接:https://developer.nvidia.com/cuda-toolkit-archive
  • 根據cuda、cudnn版本對照表下載安裝合適的cudnn
    版本對照:https://docs.nvidia.com/deeplearning/cudnn/backend/v9.12.0/reference/support-matrix.html
    下載鏈接:https://developer.nvidia.com/cudnn-archive

2. 配置conda環境

  • 去官網下載anaconda
    下載鏈接:https://repo.anaconda.com/archive
  • conda換源
    conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
    conda config --set show_channel_urls yes
    
  • pip換源
    pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/
    
  • 安裝uv
    pip install uv
    
  • 安裝conda環境(python維護周期查詢)
    conda create -n [環境名] python=3.x -y
    

3. 配置pytorch gpu環境

  • 先去官網查詢安裝torch等庫的命令
    官網鏈接:https://pytorch.org/get-started/previous-versions/
  • 使用在命令前加上uv再執行命令提高下載速度,比如
    uv pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu128
    
  • 安裝后,運行python,執行以下代碼驗證是否配置成功
    import torch
    torch.cuda.is_available()
    torch.cuda.get_device_name(0)
    

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

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

相關文章

數據處理與統計分析 —— 房源數據集分析案例

數據集網盤下載: 鏈接:https://pan.quark.cn/s/0e577858dba3?pwdFJnb 提取碼:FJnb代碼僅供參考具體可打開ipynb文件進行學習和練習:鏈接:https://pan.quark.cn/s/8efbe3061fad?pwdT47B 提取碼:T47Bimport…

藍牙如何測試?

車載藍牙測試需覆蓋 連接穩定性、功能完整性、兼容性、交互體驗等核心維度,結合車載場景的特殊性(如行駛中信號干擾、多設備交互、安全需求),具體測試點如下: 一、基礎配對與連接測試 1. 首次配對 觸發配對:車機端 “藍牙設置” 中搜索設備、手機端搜索車機(車機名稱是…

算法02 二進制與位運算

二進制作為計算機底層數據的核心表示方式,其獨特的位結構和運算規則在算法設計中有著廣泛且關鍵的應用。以下從基礎操作、算法技巧、數據結構、經典問題等多個維度,全面梳理二進制在算法中的應用: 一、基礎位運算:算法的“原子操作…

PAT 1071 Speech Patterns

題目大意是說給出一個文本,找出里面出現最多的單詞,如果有多個單詞出現次數一樣多,則輸出字典序最小的。 需要注意的是: 給出的文本字符串不僅有數字還有字母,還有一些特殊的字符,還有空格。 而單詞是只包含…

CSS中的 :root 偽類

在CSS中&#xff0c;偽類是一種用于選擇元素特定狀態的選擇器。:root 偽類專門用于選擇文檔的根元素&#xff08;在HTML中通常是<html>元素&#xff09;&#xff0c;它是CSS變量&#xff08;Custom Properties&#xff09;的理想載體&#xff0c;常用于定義全局樣式變量&…

能源行業數字化轉型:邊緣計算網關在油田場景的深度應用

能源行業數字化轉型&#xff1a;邊緣計算網關在油田場景的深度應用能源行業是國民經濟的支柱產業&#xff0c;而油田作為能源生產的重要基地&#xff0c;其數字化轉型對于提高生產效率、降低能耗、減少碳排放具有重要意義。然而&#xff0c;油田往往地處偏遠&#xff0c;油井分…

CAG緩存增強生成與RAG檢索增強生成對比

深度定制 LLM 知識,除了 RAC &#xff0c;現在又有新技術假設有一份200頁的產品手冊,你想讓 LLM 準確回答里面的相關問題,要實現這個目標,除了常用的檢索增強生成技術 rep ,現在有了新思路,緩存增強生成 CAG &#xff0c;它是什么,何時使用.RAG檢索增強是常規套路,CAG緩存增強是…

基于vue、node.js、express的網絡教學系統設計與實現/基于vue、node.js、express的在線學習系統設計與實現

基于vue、node.js、express的網絡教學系統設計與實現/基于vue、node.js、express的在線學習系統設計與實現

享元模式引發的關于ECS和對象池的思考記錄

文章目錄概念概述解決了什么區別與聯系享元模式的某個例子的細節分析概念概述 ECS&#xff08;Entity-Component-System&#xff09; 1、Entity&#xff08;實體&#xff09;&#xff1a;唯一標識符。 2、Component&#xff08;組件&#xff09;&#xff1a;純數據容器&#x…

STM32驅動SG90舵機全解析:從PWM原理到多舵機協同控制

一、SG90舵機核心特性 1.1 基本參數與選型 SG90作為??微型舵機的代表??,憑借其??輕量化設計??(僅9g)和??高性價比??,在機器人、智能小車和云臺系統中廣泛應用: ??關鍵參數對比??: ??參數?? 180定位舵機 360連續旋轉舵機 ??控制目標?? 精確…

goland怎么取消自動刪除未使用的包

1.settings-Go-Imports-取消勾選Optimize imports on the fly2.settings-Tools-取消勾選Optimize imports

halcon基于透視的可變形模型匹配

算子1&#xff0c;create_planar_uncalib_deformable_model_xld***基于平面未校準的輪廓模型算子2&#xff0c;find_planar_uncalib_deformable_model***查找平面未校準可變形模型算子3&#xff0c;projective_trans_contour_xld***將輪廓進行透視變換附加算子 算子4read_conto…

Flink Stream API - 源碼開發需求描述

概述 本文介紹如何基于Flink源碼進行二次開發&#xff0c;實現一個動態規則引擎系統。通過自定義算子和算子協調器&#xff0c;實現數據流的動態規則計算和協調管理。以此更好理解前面介紹的源碼相關文章 項目需求 核心功能 實現一個動態規則引擎&#xff0c;具備以下特性&…

「 CentOS7 安裝部署k8s」

一、Linux系統部署K8s還是非常便利的&#xff0c;只需要掌握Linux常用命令&#xff0c;便可以迅速部署&#xff0c;一起來學習一下吧1、運行以下命令更新系統并安裝必要工具&#xff1a;yum update -y yum install -y yum-utils device-mapper-persistent-data lvm22、安裝Dock…

Disbursement on Quarantine Policy(概率、逆元計算期望)

題目描述There is a train with n rows, and there are m seats per row. All seats are occupied. For some passengers, we know they are being infected with COVID-19 or not. However, for other passengers, we are not sure about their status, and we assume each of…

AI 在金融領域的落地案例

目錄 引言 一、信貸風控&#xff1a;基于 LoRA 的 Qwen-7B 模型微調&#xff08;適配城商行審批場景&#xff09; 場景背景 核心代碼 1. 環境依賴安裝 2. 金融數據集加載與預處理&#xff08;城商行信貸數據&#xff09; 3. LoRA 微調 Qwen-7B 模型 4. 模型推理&#xf…

平衡二叉樹的調整

平衡二叉樹的定義平衡二叉樹&#xff08;balanced binary tree&#xff09;&#xff0c;又稱AVL樹(Adelson-Velskii and Landis)。 一棵平衡二叉樹或者是空樹&#xff0c;或者是具有下列性質的二叉排序樹&#xff1a;① 左子樹與右子樹的高度之差的絕對值小于等于1&#xff1b;…

深入解析:如何設計靈活且可維護的自定義消息機制

深入解析&#xff1a;如何設計靈活且可維護的自定義消息機制 引言 在現代軟件開發中&#xff0c;組件間的通信機制至關重要。無論是前端框架中的組件交互&#xff0c;還是后端服務間的消息傳遞&#xff0c;一個良好的消息機制能顯著提升代碼的可維護性和擴展性。本文將深入探討…

PostgreSQL——用戶管理

PostgreSQL用戶管理一、組角色管理1.1、創建組角色1.2、查看和修改組角色1.3、刪除組角色二、角色的各種權限2.1、LOGIN&#xff08;登錄&#xff09;2.2、SUPERUSER&#xff08;超級用戶&#xff09;3.3、CREATEDB&#xff08;創建數據庫&#xff09;3.4、CREATEROLE&#xff…

東軟8位MCU使用問題總結

簡介用的單片機為ES7P7021&#xff0c;采用8位RISC內核&#xff0c;2KB的FLASH&#xff0c;128bit的RAM。編譯器使用東軟提供的iDesigner&#xff0c;開發過程中編譯器和單片機有一些地方使用時需要注意下。1.RAMclear()函數注意問題/****************************************…