我正在尝试用PyTorch Fast Dense Feature Extractor实现这篇文章,但我在将他们提供的火炬实现示例转换成PyTorch时遇到了麻烦。到目前为止,我的尝试有一个问题,即当向特征映射添加额外的维度时,卷积权重与特征形状不匹配。这在Torch中是如何管理的(从他们的实现来看,Torch似乎并不关心这一点,但PyTorch关心)。我的代码:h
PyTorch网站说,PyTorch 1.12.1与CUDA 11.6兼容,但我得到以下错误:
NVIDIA GeForce RTX 3060 Laptop GPU with CUDA capabilitysm_86 is not compatible with the current PyTorch installation.The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.我正在使用一个笔记本电脑RTX 3060