首页 > 其他分享 >Torch not compiled with Cuda enabled 解决办法

Torch not compiled with Cuda enabled 解决办法

时间:2023-10-02 13:11:42浏览次数:33  
标签:解决办法 compiled Torch enabled pytorch Cuda

  1. 确保下方指令运行有效:
nvcc --version
  1. 进入指定虚拟环境下运行下方指令:
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch


参考来源

标签:解决办法,compiled,Torch,enabled,pytorch,Cuda
From: https://www.cnblogs.com/ezhar/p/17739862.html

相关文章

  • Pytorch环境深度学习环境
    Pytorch环境深度学习环境1、安装minicoda下载地址:Miniconda—minicondadocumentation设置环境变量:安装路径\Miniconda3安装路径\Miniconda3\Scripts安装路径\Miniconda3\Library\bin测试:打开cmd,输入conda测试指令是否有效。2、配置base环境国内镜像(1)conda镜像......
  • Apex的Cuda版本与Pytorch的Cuda版本不匹配(无root权限)
    摘要(这个年代了谁还用apex啊!Pytorch自带的不好用吗?说的就是你,Megatron!)要解决这个问题,我们需要安装和Pytorch版本一致的cuda。然后再安装apex。这里简单说下步骤:下载cuda的安装程序,然后安装在无需root权限的目录。下载cudnn,把一些文件移到cuda的目录下。修改.bashrc里的环境......
  • 深度学习|如何确定 CUDA+PyTorch 版本
    对于深度学习初学者来说,配置深度学习的环境可能是一大难题,因此本文主要讲解CUDA;cuDNN;Pytorch三者是什么,以及他们之间的依赖关系。CUDACUDA(ComputeUnifiedDeviceArchitecture)是由NVIDIA开发的用于并行计算的平台和编程模型。CUDA旨在利用NVIDIAGPU(图形处理单元)的强大计......
  • 在A6000上使用rppg-toolbox 产生当前pytorch不支持sm86
    问题描述最近使用rppg-toolbox来训练一些模型,直接按照rppg-toolbox主页上的安装方式安装,在A6000主机上使用torch时产生了warning,并且程序运行十分缓慢原因产生这个问题的原因是因为按照rppg-toolbox的安装方式,安装的pytorch1.12.1是不支持CUDA11.4的,使用PiPlist查看当前安装......
  • diffusers[torch]==0.21.2 torch==2.0.1+cu117 安装失败解决办法
    安装失败主要原因是pip源使用索引为官方使用清华源设置如下pipinstall-rrequirements.txt\--extra-index-urlhttps://download.pytorch.org/whl/cu117\-ihttps://pypi.tuna.tsinghua.edu.cn/simple......
  • 如何判断torch中tensor的维度
    省流版本在输出一个张量时,你可以根据张量输出的方括号([])的数量来判断张量的维度。每个方括号对应张量的一个维度,张量的维度数就是方括号的数量。最外面的为第一个维度,往里面依次为第23...维度。例如,如果你看到如下输出:tensor([[1,2,3],[4,5,6]])这个张量有......
  • pytorch(3-0) 可视化训练误差折线图有
     缺点必须手动点击下关闭才能刷新最新的图,起码不会阻塞训练过程     ###画图训练损失训练精度测试精度importmatplotlib.pyplotaspltimportthreadingimporttimeimportmatplotlib.animationasanimationclassAnimator:def__init__(self......
  • 1、pytorch_geometric基本使用
    工具包安装方法:¶一定参考其GITHUB:https://github.com/pyg-team/pytorch_geometric(千万不要pip直接安装,肯定不行的)   In [1]:%matplotlibinlineimporttorchimportnetworkxasnximportmatplotlib.pyplotaspltdefvisualize_gra......
  • pytorch(3-2) 多层 线性回归 训练和预测代码
     脱离网页化python没有可视化    #%matplotlibinlineimporttorchimporttorchvisionfromtorch.utilsimportdatafromtorchvisionimporttransformsfromd2limporttorchasd2l#通过ToTensor实例将图像数据从PIL类型变换成32位浮点数格式,#并除以2......
  • Transformer架构解析及其pytorch实现
    备注本文对Transformer架构的分析来源于论文AttentionisAllYouNeed以及部分其引用的论文,可以理解为对该论文的翻译以及相关内容的整理。本文对Transformer的实现基于Pytorch,但是不直接调用Pytorch封装的Transformer,而是手动实现Encoder和Decoder等;与Transformer......