指定cuda版本的torch包安装

于 2024-06-12 12:58:24 发布
阅读量338 收藏 2
点赞数 3
文章标签： python YOLO 深度学习 pytorch conda
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_61033357/article/details/139623551
版权

文章目录

1.查看自己电脑的cuda版本

win+r输入cmd回车进行电脑终端界面
输入nvidia-smi指令

nvidia-smi

在这里插入图片描述

2.确定安装torch的conda指令

2.1进入网站cuda对应的torch版本

界面点击ctrl+F输入11.6（你的cuda版本如果是11.6的话），即可找到安装指令如下图
在这里插入图片描述
复制指令到所需环境的终端执行即可

3.检验torch的cuda版本是否可用

3.1终端输入python进行脚本执行界面
3.2输入 import torch
3.3输入torch.cuda.is_available()返回True表示引入成功
具体如下图
在这里插入图片描述

1. python
2. import torch
3. torch.cuda.is_available()
aloha_789
关注
  • 3
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
避坑看：最全Pytorch安装、CUDA安装
qq_44113911的博客
04-08 3349
如何正确安装Pytorch和使用conda管理不同版本
使用GPU计算python程序，如何正确安装cuda版本的torch模块，详细教程
weixin_44762762的博客
03-16 1984
使用GPU计算python程序，如何正确安装cuda版本的torch模块，从torch版本选择到模块安装，详细教程
安装特定版本的torch
weixin_44338664的博客
04-30 438
【代码】安装特定版本的torch。
Torch
doswynkfsw的博客
02-04 1万+
数据加载 torch.load()
windows：Conda、CUDA、GPU 版本的torch安装
qq_46644680的博客
03-24 4427
本文在win10环境下安装gpu版本的torch和CUDA，并测试torch
深度学习环境配置——（Cuda11）windows下的torch==1.7.1环境配置
weixin_45819759的博客
05-04 4730
好多30系显卡呀，虽然很贵，但是为了深度学习也没有办法欸。该博客主要是为了30系显卡写的，当然不一定一定要30系列显卡，只要支持都可以这样配置。
Pytorch 使用不同版本的cuda的方法步骤
09-17
总之，要在PyTorch中使用不同版本的CUDA，你需要管理好系统的CUDA库和驱动，确保Python环境中安装的PyTorch版本与CUDA版本匹配，并在编译自定义CUDA扩展时指定正确的CUDA版本。通过理解和设置这些环境变量，你可以...
pytorch安装pytorch+gpu版本安装，pytorch+cuda10.1+cudnn7.6.5安装
09-28
本教程将详细讲解如何在系统中安装PyTorch与GPU支持，特别是与CUDA 10.1和CUDNN 7.6.5配合使用的版本。 首先，了解CUDA和CUDNN是关键。CUDA是NVIDIA开发的一种并行计算平台和编程模型，它允许开发者使用GPU进行高...
conda安装cuda10.1+cudnn7.6.5，conda安装深度学习gpu环境
09-28
安装完成后，打开终端（或命令提示符），创建一个新的conda环境，指定Python版本和所需的CUDA/CUDNN版本。例如： ```bash conda create -n dl_gpu python=3.8 conda activate dl_gpu ``` 现在，我们需要安装CUDA和...
VIBE:配置环境以及论文思想与创新点分析，Ubuntu18+cuda10.1+torch-1.4.0+torchvision0.5.0
12-21
*论文： ...GitHub： ...AMASS数据集： ...** VIBE:Video Inference for Human Body Pose and Shape Estimation ** 前面的nvidia,cuda cudnn和ananconda安装百度就好 第一步创建环境 ...第二步安装几个指定版本包“ pip
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
11-27
本文将详细讲解如何在GPU环境下安装1.10.0版本的PyTorch，以及如何确保它与CUDA 10.2和11.1/11.3版本兼容。 首先，我们需要理解CUDA的重要性。CUDA是NVIDIA推出的一种并行计算平台和编程模型，它允许开发人员利用...
Linux之PyTorch安装
月生的静心苑
03-21 3295
PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。2017年1月，由Facebook人工智能研究院（FAIR）基于Torch推出PyTorch。PyTorch的前身是Torch，其底层和Torch框架一样，但是使用Python重新写了很多内容，不仅更加灵活，支持动态图，而且提供了Python接口。它是由Torch7团队开发，是一个以Python优先的深度学习框架，不仅能够实现强大的GPU加速，同时还支持动态神经网络。
gpu算力cuda版本torch版本终于搞懂了！
weixin_43603400的博客
01-25 2597
查看gpu的名称算力—在python控制台中/cmd终端nvidia-smi查看当前cuda的版本—在cmd终端stat cuda查看当前使用的cuda是什么版本的实际上就是查看/usr/local/cuda这个软链接指向的那个cuda版本安装文件nvcc -V和第一个命令一个意思。
pytoch安装指定版本教程&pytorch1.3安装笔记
不甘平凡的码农
11-07 1537
2、然后进入到以下界面，点击torch，这里会有更多的torch版本以及torchvision版本。如果电脑里安装了其他的torch版本，另外生成一个环境可以防止原先torch版本被替换掉。返回图4界面，点击torchvision进入以下界面，找到我所要的版本啦啦啦。但我安装的是cuda9.2版的torch，安装就没有这么方便了。4、在anconda终端需要联网安装，不支持提供包安装，如下图。3、终于找到了torch1.3，于是点击链接，下载安装。以上仅为个人学习笔记，如有错误，请不吝赐教，不胜感激！
查看cuda版本和torch版本
weixin_46369573的博客
09-26 2476
记录查看cuda版本和torch版本的代码。
关于我想安装cuda11.8版本的torch
qq_43900762的博客
06-28 7004
重新去搜了个命令：pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html。用这个官网的命令下，一直会把cpu版本的一起下下来，导致运行的时候一直cpu版本而不是gpu版本，torch.cuda.is_available()这个是false，问题很大。
总结：pytorch对应版本安装
热门推荐
千里之行，始于足下
04-19 8万+
为方便，自用 pytorch15 conda create -n py36_torch15_cuda92 python=3.6 conda activate py36_torch15_cuda92 conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch conda create -n py36_torch15_cuda101 python=3.6 conda activate py36_torch15_cuda1
pip install -r requirements.txt时如何安装torch的cuda版本?来看这篇文章吧
人工智能
05-01 359
pip install -r requirements.txt时如何安装torch的cuda版本? 如下命令所示，后面添加 ： -f https://download.pytorch.org/whl/torch_stable.html 即可！这样就能很好的安装我们的torch gpu版本了！
安装 PYTORCH cuda12.5
最新发布
06-10
安装 PyTorch 配合 CUDA 12.5是一个涉及多个步骤的过程，因为CUDA版本可能需要特定版本的PyTorch。下面是安装的基本步骤： 1. **检查操作系统和CUDA兼容性**： - Windows: 确保你的系统支持CUDA 12.5，查看NVIDIA官网的系统需求。 - Linux或macOS: 使用对应的操作系统包管理器或Anaconda等环境。 2. **下载CUDA**： - 访问NVIDIA开发者官网 (https://developer.nvidia.com/cuda-downloads) 下载CUDA Toolkit for your OS和CUDA版本。 3. **安装CUDA**： - 安装时遵循安装向导，注意选择合适的路径，并在安装过程中启用“Developer Tools”以便开发时调试。 4. **安装cuDNN**： - CUDA包含cuDNN，这是深度学习加速库。从NVIDIA Developer Center下载cuDNN并与CUDA版本匹配。 5. **安装PyTorch**： - 由于CUDA 12.5已经比较旧，推荐使用PyTorch的较老版本（如1.8或1.9），以防不兼容。 -f https://download.pytorch.org/whl/torch_stable.html ``` - 或者如果你想要特定版本，指定`torch==1.x.y`，例如 `pip install torch==1.9.0+cu121`，确保`cu121`代表CUDA 12.1而不是12.5。 6. **设置环境变量**： - 需要将CUDA和cuDNN的安装路径添加到系统的PATH环境变量中。 7. **验证安装**： - 运行 `nvcc --version` 检查是否安装了CUDA。 - 使用 `torch.cuda.is_available()` 来确认PyTorch是否能识别到CUDA。 **相关问题**: 1. 如何确认我的系统是否支持CUDA 12.5？ 2. 除了官方下载，还有其他渠道可以获得cuDNN吗？ 3. 如果遇到版本不兼容错误，如何解决？

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值