赞
踩
cmd中输入
nvidia-smi
查看cuda版本,我这里是12.3
直接本页面往下滑就能看到
选择与自己cuda版本相近的。
复制命令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
在自己cmd中或者pycharm终端输入即可。
不过最好自己事先安装一下numpy
pip install numpy
这样就下载好了
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。