使用的k100-AI显卡，有四张
运行BGE-M3模型，可以在cpu上跑，加载到cpu上
怎么加载到gpu上，在gpu上跑

代码为：

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from sentence_transformers import SentenceTransformer
from asgiref.sync import sync_to_async
from pydantic import BaseModel
from pymilvus import connections, Collection
from typing import Dict, List
from langchain_core.documents import Document

model = SentenceTransformer("/root/demo/model/bge/bge-m3")

##
在model = SentenceTransformer("/root/demo/model/bge/bge-m3")这一步是不是要写什么参数，类似nvidia上为cuda:0这种