Anaconda虚拟环境安装cuda和pytorch

首先电脑上要有Anaconda,使用conda创建一个虚拟环境,并激活

conda create yolov8
conda activate yolov8

win+R输入cmd,在命令窗口输入

NVIDIA-smi

可以查看到自己电脑支持的cuda环境,如下图
NVIDIA显卡相关信息
再打开torch的官网
pytorch官网
查看目前支持的版本
torch对应版本和命令
在打开的conda虚拟环境中输入

conda search cudatoolkit

有时候使用阿里源可能会查不到,如果查不到的话使用默认源应该就可以了。

查询本机支持的cudatoolkit版本,对应torch官网和NVIDIA-smi查询到当前最高支持的CUDA版本,选择一个版本的cudatoolkit和torch进行安装。
本次选择cudatoolkit=11.8

conda install cudatoolkit=11.8

如果有需要确认的输y回车或是直接回车键均可
安装完成后再安装pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

安装完成后在该环境运行python脚本进行测试

# -*- codeing = utf-8 -*-
# @Time : 2024/10/15 15:27
# @Author : 于振东
# @File : test_cuda.py
# @Software: PyCharm
import torchflag = torch.cuda.is_available()
if flag:print("CUDA可使用")
else:print("CUDA不可用")ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print("驱动为:",device)
print("GPU型号: ",torch.cuda.get_device_name(0))torch.cuda.is_available()
#cuda是否可用;torch.cuda.device_count()
#返回gpu数量;torch.cuda.get_device_name(0)
#返回gpu名字,设备索引默认从0开始;torch.cuda.current_device()
#返回当前设备索引;

效果如下

在这里插入图片描述

注:本博客为个人环境安装备忘记录,仅供参考。

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

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

相关文章

WebSocket Secure (WSS)

使用代理浏览器时,WebSocket Secure (WSS) 链接失败可能由以下原因引起: 代理设置问题: 确保代理配置正确,包括代理地址和端口。有些代理服务器不支持WebSocket连接,您需要确认您的代理服务是否支持WSS。 SSL/TLS 问题…

目标检测——Cascade R-CNN算法解读

论文: Cascade R-CNN: Delving into High Quality Object Detection (2017.12.3) 链接:https://arxiv.org/abs/1712.00726 Cascade R-CNN: High Quality Object Detection and Instance Segmentation (2019.6.24) 链接:https://arxiv.org/abs…

Z 字形变换(6)

这道题之前一直不会做,明白他是什么意思,但是找不到方法或者方法过于繁琐 方法1: 这是我在力扣评论区看到的方法,太精彩了。 虽然我实现起来效率并不高,可能是我代码的问题,但是他的思路很巧妙。 字符串的…

Spring--1

spring是一个轻量级的,采用IOC与AOP编程思想的java后端开发框架,简化了企业级的应用开发。 Spring体系 数据访问层,Web层,配置中心,测试区 IOC 控制反转,将创建对象的控制权交由Spring框架,需…

音频分割:长语音音频 分割为 短语音音频 - python 实现

在做语音任务时,有是会用到的语音音频是长音频,这就需要我们将长音频分割为短音频。 该示例将声音的音量和静默时间结合作为语音的分割条件。 使用音量和静默时间结合的分割条件,能够比较好的进行自然断句,不会话语没有说完就切断…

SpringBoot篇(一、认识SpringBoot)

目录 一、SpringBoot存在的意义 二、为什么要用SpringBoot 三、SpringBoot的优点 四、Spring & Spring Boot & Spring Cloud的关系 五、SpringBoot简化操作的体现 1. 创建独立的 Spring 应用程序 2. 嵌入式 servlet 容器 3. 提供 starter简化 Maven 配置 4. 自…

Spring声明式事务管理:深入探索XML配置方式

前言 Spring的事务管理,无论是基于xml还是注解实现,本质上还是实现数据库的事务管理机制,因此要注意发送SQL的连接是否为同一个,这是实现声明式事务的关键。 以下案例和实现基于SSM整合框架完成,不知道如何整合SSM&…

【K8S系列】Kubernetes Pod 状态详细介绍及异常状态解决方案

在 Kubernetes 中,Pod 是最小的可调度单元,负责运行一个或多个容器。Pod 的状态能够反映其生命周期中的不同阶段,帮助用户了解当前的运行状况。本文将详细介绍 Kubernetes Pod 的各种状态及其可能的异常状态解决方案。 一、Pod 状态概览 Po…

在Docker中快速使用各个版本的Oracle数据库(10g、11g、12c、18c、19c、21c、23c)

镜像地址 各个版本的Oracle数据库环境,下载地址如下: # oracle nohup docker pull registry.cn-hangzhou.aliyuncs.com/lhrbest/oracle23cfree:1.0 & nohup docker pull registry.cn-hangzhou.aliyuncs.com/lhrbest/oracle21c_ee_db_21.3.0.0 &…

查缺补漏----数据结构树高总结

① 对于平衡二叉树而言,树高的规律: 高度为h的平衡二叉树的含有的最少结点数(所有非叶节点的平衡因子均为1): n01,n11,n22 含有的最多结点数: (高度为h的满二叉树含有的结点数) ②…

Flutter在 iOS 中实现无弹窗获取剪切板内容

前言 在最新的项目需求中,我们需要在获取剪切板内容时避免弹出授权提示。这一功能是基于竞品的实现,旨在优化用户体验,特别是在推广获取跳转链接的场景下非常有用。 解决方案 通过查阅资料,我们发现对于 iOS 16 及以上的系统&a…

Fusion创建一个简单的api脚本文件

我的Fusion版本:Fusion 2.0.20476 x86_64 脚本模块在实用程序->附加模型->脚本和附加模块,快捷键为shifts 里面有一些演示脚本,可以直接使用 也可以自己创建一个新的脚本 创建的脚本在此处—— 选择脚本文件,点击编辑&a…

Unity Mirror NetworkManager初识

文章目录 Network Manager网络管理器什么是网络管理器?通过Transports进行定制化网络连接管理自定义连接地址和端口号Game State Management游戏状态管理Network Manager HUD玩家预制体及其生成控制Spawn Prefabs其他预制体注册Scene Management场景管理 Network Ma…

在Windows系统中,cmd 查看 MongoDB 相关信息

MongoDB是一种流行的NoSQL数据库,广泛应用于各种现代应用程序中。 1 查看MongoDB的版本号 要查看MongoDB的版本号,可以使用mongo命令连接到MongoDB,然后执行db.version()。 mongo连接到数据库后,执行以下命令,输出M…

读数据工程之道:设计和构建健壮的数据系统16源系统实际细节(下)

1. 数据共享 1.1. 云数据共享的核心概念是,多租户系统支持租户之间共享数据的安全策略 1.2. 任何具有细粒度权限系统的公有云对象存储系统都可以成为数据共享的平台 1.3. 数据共享也简化了数据市场的概念,在几个流行的云和数据平台上都可用 1.4. 数据…

RabbitMQ系列学习笔记(三)--工作队列模式

文章目录 一、工作队列模式原理二、工作队列模式实战1、抽取工具类2、消费者代码3、生产者代码4、查看运行结果 本文参考 尚硅谷RabbitMQ教程丨快速掌握MQ消息中间件rabbitmq RabbitMQ 详解 Centos7环境安装Erlang、RabbitMQ详细过程(配图) 一、工作队列模式原理 与简单模式相…

SpringBoot篇(二、制作SpringBoot程序)

目录 一、代码位置 二、四种方式 1. IDEA联网版 2. 官网 3. 阿里云 4. 手动 五、在IDEA中隐藏指定文件/文件夹 六、复制工程-快速操作 七、更改引导类别名 一、代码位置 二、四种方式 1. IDEA联网版 2. 官网 官网制作:Spring Boot 3. 阿里云 阿里云版制…

基于SSM+微信小程序的家庭记账本管理系统(家庭1)

👉文末查看项目功能视频演示获取源码sql脚本视频导入教程视频 1、项目介绍 1、管理员端功能有首页、个人中心、用户管理,消费详情管理、收入详情管理、系统管理等。 2、用户端功能有首页、消费详情、收入详情、论坛信息、我的等功能。 2、项目技术 …

Unity的射线检测条件

在Unity中,射线(Ray)的碰撞检测机制是当射线与添加了碰撞器(Collider)的模型发生碰撞时,射线会停止发射。这里的关键是“添加了碰撞器的模型”,也就是说,射线并不是与场景中的所有物…

django报错问题Error 0x800B0109(CERT_E_UNTRUSTEDROOT)(已解决)

报错: django.db.utils.OperationalError: (2026, ‘TLS/SSL error: Server certificate validation failed. A certificate chain processed, but terminated in a root certificate which is not trusted by the trust provider. Error 0x800B0109(CERT_E_UNTRUST…