从零开始配置pytorch环境

总结了一下几篇热门文章和自己的经验,方便以后换设备配置环境

一、安装Anaconda

这一步是为了不污染全局环境,很有必要。

直接清华源镜像下载:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

根据自己的操作系统选择合适的版本就行,不一定是最新版,个人建议选择2024.02以后的。

二、进入Anaconda Prompt

conda info -e  // 可以用来查看目前有的虚拟环境
conda create -n xxxxx python=3.9 

使用conda创建一个虚拟环境,目前只支持3.9以上版本,xxxx是你给这个虚拟环境取得名字。

conda activate xxxxx

切换进入你创建的虚拟环境,现在可以在当前环境安装pytorch了。

三、安装PyTorch

建议在官网安装,镜像容易装错版本。

(1)打开pytorch官网:PyTorch,点击Get Started

要选择pip,我装的是11.8,兼容的版本多一点,不容易出错。

把复制链接到Anaconda Prompt创建的虚拟环境中安装。

(2)安装CUDA

要想成功调用pytorch,还要装CUDA

首先查看自己的CUDA版本,win+R,输入CMD,输入以下代码查看

nvidia-smi

查看CUDA Version,我的是12.9,所以要去官网装12.9以下的版本,我装的11.8.0

直接官网装:CUDA Toolkit Archive | NVIDIA Developer

安装好一会,在Anaconda Prompt创建的虚拟环境中测试一下,依次输入以下代码:

python
import torch
torch.cuda.is_available()

结果返回true,则表示torch安装成功,并且可以调用了。如果返回false,有可能是pytorch在镜像源安装错版本了,用下面代码检查一下版本

torch.__version__

返回有cpu就是装错了,cu是对的。装错了就进下面的目录,把torch开头的文件夹都删了,重新去官网安装pytorch,也就是本文的第三步。

(3)安装好后,在pycharm配置好pytorch环境

File->settings

点击Add interpreter

conda environment会自动配置好环境,点ok就行了。

测试一下:

import torch
print(torch.__version__)           # 查看pytorch安装的版本号
print(torch.cuda.is_available())   # 查看cuda是否可用。True为可用,即是gpu版本pytorch
print(torch.version.cuda)          # 查看cuda的版本

下面内容是配置开源项目的依赖包,可以不用看

四、基于Pytorch中安装torch_geometric

在pycharm中输入以下代码,查看相关信息。

import torch
print(torch.__version__)           # 查看pytorch安装的版本号
print(torch.cuda.is_available())   # 查看cuda是否可用。True为可用,即是gpu版本pytorch
print(torch.version.cuda)          # 查看cuda的版本

返回的结果:官网安装依赖包:点击链接打开官方下载文件包网页

根据配置信息(pytorch版本是2.7.1对应2.7.0,cuda版本是11.8对应118)选择torch-2.7.0+cu118torch-2.7.0+cu118再根据自己的python版本,例如我是3.9.23,对应的就是cp39,除了lib不下载,其他四个包都要下载。

例如,torch scatter我下载的就是​​​​​​​​torch_scatter-2.1.2+pt27cu118-cp39-cp39-win_amd64.whl​

下载后,进入Anaconda Prompt创建的虚拟环境中安装

pip install torch_scatter xxxx
pip install torch_sparse xxxx
pip install torch_cluster xxxx
pip install torch_spline_conv xxxx

xxxx表示你这个依赖包的地址和依赖包名字,例如:

pip install torch_scatter E:\Anaconda\Scripts\torch_scatter-2.1.2+pt27cu118-cp39-cp39-win_amd64.whl

四个依赖包安装后,就可以安装torch-geometric

pip install torch-geometric

参考文献:

​​​​​​​在Pycharm中安装Pytorch教程总结_pycharm安装pytorch-CSDN博客

torch.cuda.is_available()返回false——解决办法_cuda available false-CSDN博客

基于Pytorch中安装torch_geometric简单详细完整版-阿里云开发者社区

完美解决AttributeError: module ‘numpy‘ has no attribute ‘typeDict‘_module 'numpy' has no attribute 'typedict-CSDN博客

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:http://www.pswp.cn/pingmian/91794.shtml
繁体地址,请注明出处:http://hk.pswp.cn/pingmian/91794.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

离线录像文件视频AI分析解决方案

前言 在安防领域,视频的价值不言而喻,尤其是录像文件的价值显得更是尤为重要。在一个安防项目上视频录像一般存储1个月、3个月甚至更长时间,这就形成了海量的离线录像文件数据。传统安防项目对录像文件的处理还有一定的局限性: 事…

人工智能与金融:金融服务的重塑

1. AI 驱动金融变革:从 “标准化” 到 “智能化”1.1 传统金融的痛点与 AI 的破局传统金融行业长期面临三大瓶颈:服务效率低下:银行柜台办理一笔贷款需 3-5 个工作日,人工审核流程繁琐,客户等待时间长;风险…

Qwen大模型关键参数解析:缓存与滑动窗口

QwenLLM "use_cache": true Qwen "use_sliding_window": false 在大语言模型(如Qwen系列)的生成配置中,"use_cache": true 和 "use_sliding_window": false 是两个与模型推理效率和长文本处理相关的重要参数,具体含义如下: 1. …

Flask 框架全面详解

Flask 是一个轻量级的 Python Web 框架,以其简洁、灵活和易扩展的特性广受欢迎。接下来我将从多个维度详细介绍 Flask 框架。 1. Flask 核心特性 微框架设计 轻量级:核心功能精简,只包含基本组件可扩展:通过扩展添加所需功能无…

概率 多维随机变量与分布

一、二维1、二维随机变量及其分布假设E是随机试验,Ω是样本空间,X、Y是Ω的两个变量;(X,Y)就叫做二维随机变量或二维随机向量。X、Y来自同一个样本空间。联合分布函数 F(x,y)P(X≤x,Y≤y),即F(x,y)表示求(x,y)左下方的面积。 F(x,…

Spring AI MCP:解锁大模型应用开发新姿势

一、AI 浪潮下的新利器 ——Spring AI MCP 登场在当今数字化时代,人工智能(AI)无疑是最耀眼的技术明星,正以前所未有的速度渗透到各个领域,深刻改变着我们的生活和工作方式。从智能语音助手到图像识别技术,…

ThinkPHP5x,struts2等框架靶场复现

ThinkphpThinkphp5x远程命令执行及getshell首先我们先找一个环境,或者自己搭建一个环境fofa:body"ThinkPHP V5"搭建:vulhub/thinkphp/5-rcedocker-compose up -d然后去访问我们的环境远程命令执行/?sindex/think\app/invokefuncti…

Hyperliquid:揭秘高性能区块链共识引擎HyperBFT

大家好,今天我们一起探讨Hyperliquid这个高性能区块链项目以及它背后的共识引擎。Hyperliquid能在拥挤的去中心化交易赛道(DEX)中脱颖而出,很大程度上要归功于其高效的共识机制——HyperBFT。 为了彻底搞懂HyperBFT,我…

大模型开发框架LangChain之构建知识库

1.前言 为了避免 llm正确的废话和幻觉,知识库可以说是现在开发 agent的必备了。同时,作为 rag中的 r,知识库召回的成功率会极大的影响 llm的最终回复效果。一般,会把知识库召回的内容作为背景知识给到 llm,并在 prompt…

NPM打包时,报reason: getaddrinfo ENOTFOUND registry.nlark.com

先说解决方法:将 package-lock.json 文件中的 registry.nlark.com 改为 registry.npmmirror.com现象:npm ERR! code ENOTFOUND npm ERR! syscall getaddrinfo npm ERR! errno ENOTFOUND npm ERR! network request to https://registry.nlark.com/url-too…

python内置库os与sys的区别是什么?分别能实现什么功能?

Python 的 os 和 sys 是两个功能截然不同但都非常重要的内置库,它们分别服务于不同的交互场景:前者专注于与操作系统交互,后者专注于与 Python 解释器本身交互。以下是详细解析: 一、os 库:操作系统交互接口 os 库&…

【云计算】云主机的亲和性策略(一):快乐旅行团

《云主机的亲和性策略》系列,共包含以下文章: 1️⃣ 云主机的亲和性策略(一):快乐旅行团2️⃣ 云主机的亲和性策略(二):集群节点组3️⃣ 云主机的亲和性策略(三&#xf…

USRP捕获手机/路由器数据传输信号波形(下)

目录: USRP捕获手机/路由器数据传输信号波形(上) USRP捕获手机/路由器数据传输信号波形(中) USRP捕获手机/路由器数据传输信号波形(下) 四、路由器MIMO-OFDM系统 本文深入分析采集手机与路由…

位运算在权限授权中的应用及Vue3实践

在现代前端应用中,权限管理是一个至关重要的功能模块。随着应用复杂度的提示功能,权限细粒度越来越精细,如何高效地管理和判断权限成为前端开发的一大挑战。位运算作为一种高效的运算方式,在权限管理领域有着独特的优势。本文将详…

面试实战,问题二十二,Java JDK 17 有哪些新特性,怎么回答

Java JDK 17 新特性面试回答指南 作为一名Java开发者,了解JDK 17的新特性是面试中的关键点。JDK 17(Java SE 17)于2021年9月发布,是一个长期支持(LTS)版本,引入了多项改进以提升开发效率、安全性…

【MySQL安全】什么是SQL注入,怎么避免这种攻击:前端防护、后端orm框架、数据库白名单

基本概念SQL注入是OWASP Top 10安全风险之一,它利用了应用程序对用户输入数据的不当处理。当应用程序直接将用户输入拼接到SQL查询中而没有进行适当的过滤或转义时,就可能发生SQL注入攻击。攻击原理假设有一个登录表单的SQL查询:SELECT * FRO…

pyqt5显示任务栏菜单并隐藏主窗口,环境pyqt5+vscode

环境 pyqt5vscode 环境搭建见 https://blog.csdn.net/huiaifen/article/details/125175261 新建一个QMainWindow 1 在VSCode的资源管理器中,右键选择 PYQT:New Form,打开Qt Designer2 在打开的窗口中选 “Main Window”,然后选“创建”3 直接…

SpringBoot项目数据脱敏(自定义注解)

文章目录前言一.配置1.脱敏类型枚举:DesensitizeType2.注解:Desensitize3.序列化类:DesensitizeJsonSerializer4.工具类:DesensitizeUtil二、测试:DesensitizeTest三、效果展示总结前言 在互联网应用中,用户…

PSO-TCN-BiLSTM-MATT粒子群优化算法优化时间卷积神经网络-双向长短期记忆神经网络融合多头注意力机制多特征分类预测/故障诊断Matlab实现

基本介绍 1.Matlab实现PSO-TCN-BiLSTM-MATT粒子群算法优化时间卷积神经网络-双向长短期记忆神经网络融合多头注意力机制多特征分类预测,PSO-TCN-BiLSTM-Multihead-Attention; 多头自注意力层 (Multihead-Self-Attention):Multihead-Self-Atte…

第一篇:Linux 运维入门:虚拟机部署与基础环境配置

目录 一、准备工作与环境规划 二、虚拟机网络配置 1、虚拟网络编辑器设置 2、系统网络配置 3、主机名配置 三、Hosts 文件与 SSH 免密配置 配置 hosts 文件编辑/etc/hosts文件实现主机名解析: 分发 hosts 文件到其他节点 SSH 免密登录配置在 zhangsan101 上…