win11 安装pytorch

发布时间 2023-12-17 20:50:36作者: dmesg

第一步骤,是安装cuda和cudnn,看这个链接。 https://zhuanlan.zhihu.com/p/586913250。重点就是如何安装pytorch了,因为网络慢的原因

 

 

首先,直接使用下面这个命令不好使,如果你的科上网流量非常足的话,没问题的。我用aconda,差不多就安装好了,可是比较可惜的是,机器内存不够(32G),aconda安装不了,只是删除掉,使用了miniaconda了

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

1,科上网流量太贵,上面方法不太可行

2,完全使用清华的源,但是有个问题,指定了cuda版本后,清华的包不是最新的,有些版本没有。如果不指定cuda版本的话,装的是cpu版本,如下面的命令
conda install pytorch torchvision torchaudio

不过清华的源,应该也是可行,不过你得先看看他的的源上有那些cuda的版本,然后你去对应的装对应的版本,如果你先装cuda,尤其是我这种的情况的,再去找pytorch的源就麻烦了,他缺少最新的版本。

3,,就是下载whl,使用pip来安装了,我之前一直反感使用pip来安装,因为如果出现动态库的问题,尤其是之前安装tf的时候,这次只能试试了,没想到最终成功了。
具体就是在下面的链接找到具体的地址,然后下载到本地,使用pip install来安装
https://download.pytorch.org/whl/torch/


我下载是

torch-2.1.0+cu121-cp39-cp39-win_amd64
torchvision-0.16.0+cu121-cp39-cp39-win_amd64.whl
torchaudio-2.1.0+cu121-cp39-cp39-win_amd64.whl