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
这样就下载好了