安装cuda版本的pytorch

先下载cuda版本的pytorch的整个打包文件:

pip install torch-2.5.1+cu124-cp312-cp312-linux_x86_64.whl

验证:

#python
import torch
torch.__version__

posted on 2025-01-12 11:05 paulwong 阅读(12) 评论(0) 编辑收藏所属分类: AI-PYTHON


只有注册用户登录后才能发表评论。




网站导航: 博客园 IT新闻 Chat2DB C++博客博问管理
相关文章: 安装cuda版本的pytorch python界面库

paulwong