PyTorch是一个深度学习框架,它实现了一个动态计算图,它允许您改变神经网络在运行中的行为方式,并能够执行向后自动区分。
我正在尝试使用使用火炬的脚本,但我不断收到此属性错误: AttributeError:部分初始化的模块“torch”没有属性“Tensor”(很可能是由于循环 im...
我找不到 CUDA 12.4 的 pytorch 和 cudnn 版本。我可以下载和使用哪些版本?
nvidia-smi 输出显示 CUDA 12.4,如下所示。 我下载并安装了它作为 CUDA 工具包 并下载了 cudnn 第一个: 12.4 没有选择。因此我下载了 pytorch...
如何将torch集成到docker镜像中,同时保持镜像大小合理?
所以我有一个 Flask Web 应用程序,它将公开一些深度学习模型。 我构建了图像,一切正常。 问题是这张图片的大小是5.58GB!这有点荒谬。 我...
是否有 Pytorch 内部程序来检测张量中的 NaN? Tensorflow 有 tf.is_nan 和 tf.check_numerics 操作...Pytorch 在某处有类似的东西吗?我找不到
`ptxas 警告:Tensorflow 和 PyTorch 上的寄存器溢出到本地内存
在我们的一项研究中,我们将 Tensorflow 和 Pytorch 与其他主要模型一起使用。每当我们在大学使用数据服务器时,我们都可以在模型训练中使用完整的 GPU。电脑上有一个
如何在Windows 10上安装支持CUDA的pytorch 1.2.0?
我尝试在 Windows 10 上使用 Python 3.9 安装支持 CUDA 的 pytorch 1.2.0,但一直出错。 例如,我 pip install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c py...
我使用 PyTorch 实现了一个简单的线性回归模型,并尝试使用 GPU 加速训练。尽管如此,我并没有观察到训练时间有任何改善。剖析 mo...
如何在针对移动设备优化的模型上启用 Pytorch Dropout
我有一个使用 dropout 进行 MC 预测的模型。它在我的桌面上的 Python 中运行良好。我想在移动平台上执行相同的算法,但 dropout 似乎没有应用。 ...
我设计了一种新的注意力机制。让我们用一句话来说“打篮球太酷了!”我有一对玩篮球的代币。这两个标记有一些嵌入向量,对于
如何基于另一个张量创建张量 - 在实践中学习 PyTorch?
我正在使用 PyTorch 学习 IA 并实现一些玩具示例。 首先,我创建了一个一维张量 (X) 和第二个张量 (y),源自第一个张量: X = torch.arange(0, 100, 1.0).
何时在 pytorch Lightning 中使用prepare_data 与 setup?
Pytorch 关于数据加载器的文档仅在代码中说 def 准备数据(自身): # 下载 ... 和 def setup(self, stage: 可选[str] = None): # 分配训练/验证数据集以在
如何在PyTorch Lightning中从prepare_data()获取数据集到setup()
我使用PyTorch Lightning的DataModules方法在prepare_data()方法中使用NumPy制作了自己的数据集。现在,我想将数据传递到 setup() 方法中以分为训练和
我是 PyTorch 的新手,想要制作一个带有 255x255 RGB 图像的简单自动编码器来使用它,但是输出形状与输入形状不同。 这是模型
我正在尝试迭代初始化如下的 pytorch 数据加载器: trainDL = torch.utils.data.DataLoader(X_train,batch_size=BATCH_SIZE, shuffle=True, **kwargs) 其中 X_train 是 pandas 数据...
我需要将一个普通的 numpy 数组与一个空数组相加 x = np.ones([2,3]) x + np.array([]).reshape(2,-1) 输出: ValueError:操作数无法与形状 (2,3) (2,0) 一起广播 雷什...
我有一些时间序列(以 0.5s 和 3kHz 采样,这意味着每个时间序列都有 1,500 个点),想要通过神经网络进行二元分类。序列看起来像照片...
我想匹配列表中张量的索引。 我正在尝试使用 Pytorch 进行链接预测。 在这个过程中,我需要通过将索引映射到字典来将其转换为名称。 要做...
我需要用我训练的卷积神经网络的数据测试结果编写一个文件。数据包括语音数据采集。文件格式需要是“文件名、预测...
我有一个代码,其中大小为 (20, 20) 的 2D 网格在展平 (400) 时需要根据 2D 网格上的所有其他索引计算距离。目前,我正在使用 for 循环来存储它。 # 最佳定位...
如何指定 model.learn() 在稳定基线 3 的特定片段内结束?
我知道指定total_timesteps=是一个必需参数,但是如何在特定的episode内结束model.learn()?请原谅我,因为我对 stables_baselines3 还很陌生,而且 pytorch 还不太熟悉...