为 Transformer 的 GPU 训练安装 Spacy

Question

我还安装了 CudaToolkit 和 cuDNN。

我有以下内容

GPU:
| NVIDIA-SMI 531.68                 Driver Version: 531.68       CUDA Version: 12.1     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                      TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce RTX 3070 T...  WDDM | 00000000:01:00.0 Off |                  N/A |
| N/A   43C    P8                9W /  N/A|      0MiB /  8192MiB |      0%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+

+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|  No running processes found                                                           |
+---------------------------------------------------------------------------------------+

我一直在尝试按照以下说明在网站上的几个帖子上安装 spacy 并使用

pip install spacy[cuda121]

我收到一条警告，spacy 3.5.2 未检测到或不支持 CUDA 12.1 版。

我设法让 tensorflow 检测 GPU，当我运行以下命令时：

import tensorflow as tf
print('Num GPUs Available: ', len(tf.config.list_physical_devices('GPU')))

它检测到 1 个 GPU。 我也跑了：

pip install cudy

我想知道是不是设置有问题或者我做错了什么。任何支持将不胜感激。