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

Anacond虚拟环境里安装GPU版本Pytorch

CUDA/cudnn/CUDAToolkit简单区分

pytorch与cuda关系简单介绍

pytorch是一个深度学习的框架,能够调用cuda进行gpu加速。一般来说你的电脑里是有相关cuda的驱动的(可用nvidia-smi命令验证,能运行就说明有,不能运行去搜索安装),所以要在anaconda虚拟环境里安装pytorch就不用安装cuda了

安装教程

一、查看电脑CUDN版本

在cmd(windows)或者终端(linux)输入命令:

nvidia-smi

下图是我电脑的示例

其中CUDA Version后是指最高支持的cuda版本,所以要安装的pytorch能够调用的cuda不能超过12.8!!!

我这以安装12.1版本为例

二、进入pytorch官网:

官网链接如下:

PyTorch https://pytorch.org/

        进入官网后直接下滑找到如下位置、并选择对应的内容

如果有你想要的CUDA版本,这里不建议使用conda命令进行安装(不仅慢还容易安装到cpu版本的),建议使用离线下载,将安装包下载到本地电脑,这样以后其他地方有需要可以直接安装。请在Package这里选择pip如下图

请注意,我们最终的目的,就是要安装上图用红框圈出来的torchtorchvisiontorchaudio这三个包。

有些人会发现了,这里的CUDA11.8、CUDA12.4太低或太高了,而我想要安装个CUDA 12.1,或者我想安装个其他版本的,那怎么办呢?如果是太高了那我们可以进入网址中选择对应版本的CUDA。不要急,请看官网对应的下面这张图的位置

1.点击上面序号(1)内的跳转,进入如下页面查找自己电脑CUDN对应的版本号;以及torchtorchvisiontorchaudio这三个分别对应的版本号。

或此链接直达:

Previous PyTorch Versions | PyTorchhttps://pytorch.org/get-started/previous-versions/

2.复制上图序号(2)中绿框中的链接,在浏览器进入绿框(2)圈出来的地址(就是上图pip命令最后给的网址),就是在这个地址,我们要找到并下载上述的这三个包(torchtorchvisiontorchaudio)。请注意,一定要下载带"+cu"字样的、与你python版本一致的、与你的操作系统一致的包。

或此链接直达
离线安装位置http://download.pytorch.org/whl/点击链接后进入,如下图所示位置。

三、下载离线包

我想要安装cuda 12.1的、windows平台下的、python版本为3.8的包,包下载到哪里都可以,最终下载完的三个包如下图

然后,打开cmd或者终端,激活你想要安装pytorch虚拟环境

激活完环境后,运行命令"pip install 包的完整路径"。依次安装完这三个包就好啦。

比如我想要在我建立的“py38”环境下安装pytorch

四、在conda虚拟环境中安装torch

# 激活环境

conda activate py39

# 安装包,我的下载上述3个包在D:\A-programming\A-Python\GPU\cuda12.1\这个路径下的,然后依次在终端中使用以下命令安装即可。

pip install "D:\A-programming\A-Python\GPU\cuda12.1\torch-2.3.0+cu121-cp38-cp38-win_amd64.whl"

pip install "D:\A-programming\A-Python\GPU\cuda12.1\torchvision-0.18.0+cu121-cp38-cp38-win_amd64.whl"

pip install "D:\A-programming\A-Python\GPU\cuda12.1\torchaudio-2.3.0+cu121-cp38-cp38-win_amd64.whl"

五、验证是否安装成功

新建一个python文件,写上以下代码

import torch

print(torch.cuda.is_available())

运行,如果输出True,那就是成功了。

http://www.dtcms.com/a/121933.html

相关文章:

  • linux下截图工具的选择
  • Linux权限管理:从入门到实践
  • 网页部署到宝塔服务器上,发送请求报错?org.springframework.data.redis.RedisSystemException,让我来看看
  • 算法训练之动态规划(一)
  • 为什么在Altium原理图中画线会自动链接到附近?
  • Zephyr、FreeRTOS、RT-Thread 定时器区别分析
  • 【Ragflow】14.MinerU解析脚本,接入ragflow知识库
  • Windows 图形显示驱动开发-WDDM 2.0功能_供应和回收更改
  • Seq2Seq - GRU补充讲解
  • DP_AUX辅助通道介绍
  • Testbench设计
  • 起跑有几种起跑方式·棒球1号位
  • 【嵌入式系统设计师】知识点:第11 章 嵌入式系统设计案例分析
  • MQ保证消息幂等
  • (四十一)Dart 中的空安全与 `late` 关键字教程
  • 网络基础-路由技术和交换技术以及其各个协议
  • 每日算法-250409
  • 智能体:重塑未来的智能助手及其在产品设计中的应用
  • 逻辑分析仪怎么用?怎么看PWM(以Saleae为例)​
  • TQTT_KU5P开发板教程---DDR4基于IP核 SDRAM(MIG)的测试
  • 接口测试性能指标:QPS、TPS、OTPS、TP99
  • 【深度学习基础】Temporal Fusion Transformer
  • TCP转发隧道
  • Go并发背后的双引擎:CSP通信模型与GMP调度|Go语言进阶(4)
  • K8S学习之基础七十九:关闭istio功能
  • 【c语言】倒置字符串
  • map/multimap
  • vue3项目中实现国际化语言切换【直接粘贴即可】
  • Windows找不到compiler.automake.allow.when.app.running怎么设置?
  • ArcGIS Engine开发教程--从零搭建GIS桌面应用