17427 1 year ago
parent 755929eac1
commit cf2a588b5e

@ -0,0 +1,5 @@
# 首先卸载
conda uninstal torch torchvision torchaudio -y
# 然后安装
conda install pytorch==2.1.0+cu121 torchvision==2.1.0+cu121 torchaudio==2.1.0+cu121 pytorch-cuda=12.1 -c pytorch -c nvidia
Loading…
Cancel
Save