时隔几个月,再次重回炼丹领域,还是啥也不会。。。
不知道什么原因,原本的torch1.6.0+cu101(好像是101忘记了)出现了莫名其妙的问题,所有关于torch.nn下的函数和包全部报错,只好重装。
- 查询本机的cuda版本.
- 登录pytorch官网,选择对应版本的torch进行安装,如笔者的版本是11.1,就安装对应的版本。pytorch官网因为使用的pip安装,所以直接复制下面的文本,在cmd中进行安装即可。
- 在安装中出现了几个莫名其妙的错误,①zipFile Badzipfile,没找到具体的原因,可能是因为我之前安装过torch和torchvision,卸载的时候只卸载了torch,后来把torchvision也卸载了就再也没出现该错误;②ReadTimeoutError: HTTPSConnectionPool,后来在安装时额外添加了
--default-timeout=100
就没问题了,完整的安装代码类似pip3 --default-timeout=100 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html