torch的下载
在自己cmd中或者pycharm终端输入即可。查看cuda版本，我这里是12.3。不过最好自己事先安装一下numpy。选择与自己cuda版本相近的。直接本页面往下滑就能看到。
·
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
这样就下载好了
更多推荐
13
0- 0
已为社区贡献3条内容
所有评论(0)