AMD AI 开发者计划

通过魔搭账号登录后新人赠送10小时云算力

进入jupyter后,新建一个终端,输入以下命令检查显卡情况

amd-smi

并检查pytorch是否识别AMD显卡

python -c "import torch; print('PyTorch:', torch.__version__); print('ROCm available:', torch.cuda.is_available()); print('Device:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'N/A')"

在云服务器环境中安装魔搭ModelScope,安装后输入命令安装Gemma4

modelscope download --model google/gemma-4-E4B-it --cache_dir "./models"

卸载重装torchvision后确保vLLM服务为最新,然后启动服务

vllm serve ./models/google/gemma-4-E4B-it/ --served-model-name gemma-4-E4B-it

这里需要等待一会,直到出现 Application startup complete 就可以下一步了

打开一个新终端,输入命令

vllm chat --url http://localhost:8000/v1 --model gemma-4-E4B-it

等到出现

>

就可以进行对话了

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐