当前位置:   article > 正文

torch的下载_torch下载

torch下载

cmd中输入

nvidia-smi

查看cuda版本,我这里是12.3 

然后进入pytorch官网PyTorch

直接本页面往下滑就能看到 

选择与自己cuda版本相近的。

复制命令:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

 在自己cmd中或者pycharm终端输入即可。

不过最好自己事先安装一下numpy

pip install numpy

这样就下载好了 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/458831
推荐阅读
相关标签
  

闽ICP备14008679号