安裝cuda版本的pytorch
先下載cuda版本的pytorch的整個打包文件:pip install torch-2.5.1+cu124-cp312-cp312-linux_x86_64.whl
驗(yàn)證:
#python
import torch
torch.__version__
import torch
torch.__version__
posted on 2025-01-12 11:05 paulwong 閱讀(74) 評論(0) 編輯 收藏 所屬分類: AI-PYTHON