今天重新配环境的时候要安装torch的cuda12.4版本,我尝试了好多方法,其中官网的指令下载实在太慢、清华源镜像配置后下载依旧缓慢(maybe是我操作不对,但我不想再试了),于是乎,我就直接先在终端敲入官网的那个指令

然后找到了那个轮子.whl文件下载地址:https://download.pytorch.org/whl/cu124/torch-2.4.0%2Bcu124-cp38-cp38-win_amd64.whl

点击这个后直接下载,不过这时候需要挂个vpn加速一下,速度我这边是几MB/s,还可以不算慢,十分钟内可以下完,下载好后找到路径,然后直接在终端pip安装:

这个安装好后会显示:

找了一下发现没下载torchvision和torchaudio,然后我分别用pip单独下载了一下(注意我没指定版本,但cuda12.4版本可能比较新,所以我没指定版本可能误打误撞对了,正常情况下三者的版本是要对应的),搞好后我试了一下,应该是成功了:

再用pip list确定一下:

希望能帮到大家!

---------------------------------------------------------------------------------------------------------------------------------更新一下:torchvision和torchaudio可以把官网的指令拷贝下来(去掉torch即可)

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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐