指定cuda版本的torch包安装

于 2024-06-12 12:58:24 发布
阅读量356 收藏 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的方法步骤
09-17
总之，要在PyTorch中使用不同版本的CUDA，你需要管理好系统的CUDA库和驱动，确保Python环境中安装的PyTorch版本与CUDA版本匹配，并在编译自定义CUDA扩展时指定正确的CUDA版本。通过理解和设置这些环境变量，你可以...
torch.cuda.is_available（）返回False解决方案
12-21
可以通过访问PyTorch官网或使用指定CUDA版本的安装命令来解决这个问题。例如，对于CUDA 10.0，可以使用如下命令安装： ``` pip install ...
一站式安装对应显卡版本的cuda和torch（windows）
Pan_peter的博客
02-16 3246
1、安装python环境2、安装VS的C++环境（因为CUDA在安装时，需要VS的里面的工具包来编译。3、安装cuda4、安装pytorch。
避坑看：最全Pytorch安装、CUDA安装
qq_44113911的博客
04-08 3840
如何正确安装Pytorch和使用conda管理不同版本
使用GPU计算python程序，如何正确安装cuda版本的torch模块，详细教程
weixin_44762762的博客
03-16 1995
使用GPU计算python程序，如何正确安装cuda版本的torch模块，从torch版本选择到模块安装，详细教程
Torch
热门推荐
doswynkfsw的博客
02-04 1万+
数据加载 torch.load()
windows：Conda、CUDA、GPU 版本的torch安装
qq_46644680的博客
03-24 4526
本文在win10环境下安装gpu版本的torch和CUDA，并测试torch
深度学习环境配置——（Cuda11）windows下的torch==1.7.1环境配置
weixin_45819759的博客
05-04 4848
好多30系显卡呀，虽然很贵，但是为了深度学习也没有办法欸。该博客主要是为了30系显卡写的，当然不一定一定要30系列显卡，只要支持都可以这样配置。
Linux之PyTorch安装
月生的静心苑
03-21 3325
PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。2017年1月，由Facebook人工智能研究院（FAIR）基于Torch推出PyTorch。PyTorch的前身是Torch，其底层和Torch框架一样，但是使用Python重新写了很多内容，不仅更加灵活，支持动态图，而且提供了Python接口。它是由Torch7团队开发，是一个以Python优先的深度学习框架，不仅能够实现强大的GPU加速，同时还支持动态神经网络。
gpu算力cuda版本torch版本终于搞懂了！
weixin_43603400的博客
01-25 2774
查看gpu的名称算力—在python控制台中/cmd终端nvidia-smi查看当前cuda的版本—在cmd终端stat cuda查看当前使用的cuda是什么版本的实际上就是查看/usr/local/cuda这个软链接指向的那个cuda版本安装文件nvcc -V和第一个命令一个意思。
官网使用conda&pip安装PyTorch命令总结（包含各版本）
MrNeo的博客
04-23 3778
因为有时访问该网站比较慢，所以本博客记录该网页内容。
pytoch安装指定版本教程&pytorch1.3安装笔记
不甘平凡的码农
11-07 1583
2、然后进入到以下界面，点击torch，这里会有更多的torch版本以及torchvision版本。如果电脑里安装了其他的torch版本，另外生成一个环境可以防止原先torch版本被替换掉。返回图4界面，点击torchvision进入以下界面，找到我所要的版本啦啦啦。但我安装的是cuda9.2版的torch，安装就没有这么方便了。4、在anconda终端需要联网安装，不支持提供包安装，如下图。3、终于找到了torch1.3，于是点击链接，下载安装。以上仅为个人学习笔记，如有错误，请不吝赐教，不胜感激！
安装特定版本的torch
最新发布
weixin_44338664的博客
04-30 474
【代码】安装特定版本的torch。
查看cuda版本和torch版本
weixin_46369573的博客
09-26 2666
记录查看cuda版本和torch版本的代码。
关于我想安装cuda11.8版本的torch
qq_43900762的博客
06-28 7314
重新去搜了个命令：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，问题很大。
安装 torch
weixin_54959970的博客
03-18 1281
pytorch 的安装
cuda和torch安装
09-08
对于CUDA和Torch的安装，可以按照以下步骤进行： 1. CUDA的安装： - 首先，检查您的显卡是否支持CUDA，并确保您有合适版本的显卡驱动程序。您可以在NVIDIA网站上找到有关支持的版本和驱动程序的信息。 - 下载适用于您操作系统的CUDA安装包，并按照指示进行安装。可以从NVIDIA官方网站下载。 - 安装完成后，将CUDA的安装目录添加到您的系统环境变量中。 2. Torch的安装： - 首先，确保您已经安装了Python，并且已经安装了适合您系统的Python包管理工具（如pip或conda）。 - 在命令行窗口中运行以下命令，使用pip安装Torch： ``` pip install torch ``` 或者使用conda安装： ``` conda install pytorch torchvision torchaudio cudatoolkit=<CUDA版本号> ``` 如果您希望安装特定版本的Torch，请在命令中指定版本号。 3. 安装完成后，通过导入torch模块来验证Torch是否正确安装： ```python import torch print(torch.__version__) ``` 如果没有报错并且正确显示了Torch的版本号，则表示已成功安装。 请注意，根据您的操作系统、CUDA版本和其他依赖项的不同，上述步骤可能会有所不同。因此，请查阅官方文档以获取更详细的安装说明。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值