vision transformer硬件
时间: 2023-09-30 18:05:22 浏览: 69
视屏转换器
根据提供的引用内容,Vision Transformer的硬件要求包括:
- 使用到了MultiHeadAttention,对应tensorflow-gpu最低版本为2.4.0。
- 安装tensorflow-gpu版本需要使用cudatoolkit=11.0和cudnn==8.0.5.39。
因此,为了使用Vision Transformer,您需要具备以下硬件条件:
- 支持CUDA Toolkit 11.0的显卡。
- 安装了对应版本的cuDNN库。
- 安装了tensorflow 2.4.0及以上的GPU版本。
请注意,这只是Vision Transformer所需的硬件条件之一,还需要考虑其他系统和软件要求。
阅读全文