也不知道为啥torch用镜像网站不能下,难道镜像网站没这个?
但是有呀,就很迷。
文章目录
- 1.pytorch官网直接命令下载
- 2.到官网下载wheel文件离线下载
- 对pytorch进行测试
1.pytorch官网直接命令下载
- 进入pytorch官网:官网链接
- 选择合适的版本获得命令
- 在终端输入改命令即可下载
有可能会速度过慢导致中断,再用下面的方法??。
1.1还是试了试镜像网站pip,但是不行,然后在网上找了找方法,但好多都是conda的,没安装anaconda没有排面了(/(ㄒoㄒ)/~~)。
1.2.也试了试在镜像网站把Pytorch和torchvision的压缩包下载离线下载
感觉我把包都下下来了!应该总没问题了吧!但,嘿嘿,意料总是会来的,会提示
1 | ERROR: Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output |
这玩意的错,如果往前看,能看到下面的错误
1 2 3 | ModuleNotFoundError: No module named 'tools.nnwrap' ---------------------------------------- ERROR: Failed building wheel for torch |
编译错误的话,说明没有相应的torch wheel 文件,所以我们要下载相应的wheel文件!
2.到官网下载wheel文件离线下载
回到官网,打开命令中的链接
然后再这个网址上下载wheel文件,如下
找到自己所对应的torch版本,进行下载(tips:迅雷下载超快的哦!尽量别再晚上下载,限速!!晚上太亏了)
只需要下载torch就好了哦,torchvision等下可以pip很快的。
下载完成过后,打开命令行,cd到刚刚的安装目录进行pip install 【相应的wheel文件】。
安装完成过后,我直接pip install torchvision不行…所以!又回到官网!!
用官网下方的命令行!因为已经安装好torch,所以很快就安装好相应的torchvision版本啦!!
酱酱就安装好了。
对pytorch进行测试
1 2 3 4 5 6 7 8 9 10 11 12 13 | import torch print(torch.__version__) print(torch.cuda.is_available()) x = torch.randn(1) if torch.cuda.is_available(): device = torch.device("cuda") y = torch.ones_like(x, device=device) x = x.to(device) z = x + y print(z) print(z.to("cpu", torch.double)) |
运行以上代码,若能正常输出就可以正式展开代码的世界了。