Pytorch(GPU)安装小结 |
您所在的位置:网站首页 › pytorch作用 › Pytorch(GPU)安装小结 |
引:最近在学习神经网络的搭建与使用,需要安装Pytorch,但是在安装的过程中遇到了很多问题,在这里总计一下。 1.国内镜像源众所周知(手动狗头),Python的好多库是需要翻墙访问外网进行下载的,但是翻墙的速度。。。。。。贼拉慢,所以要配国内的镜像。我搜了浏览最高的一堆帖子之后,最后还是换回了清华的镜像。(因为其他几个总是报错) 贴上我anaconda的condarc文件的镜像配置,不知道怎么改镜像的同学可以去其他dai佬那里看一下,我就不献丑了。 channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ - defaultsshow_channel_urls: true2.无法调用GPU(1)cpuonly库根据官网给的代码下载Pytorch之后,验证是否能调用GPU import torchprint(torch.__version__)print("gpu", torch.cuda.is_available())很遗憾,使用清华镜像源的是不能调用GPU的,因为使用清华的镜像源在下载的时候会自动下载cpuonly库,这个库会限制只使用cpu而不调用GPU,找到这个库卸载掉就可以 如果查询cpuonly并不存在,(骚操作)此时要先安装cpuonly库然后在卸载,这个过程中会重新安装一个适配的pytorch库(应该是GPU版的) 装完太高兴直接把页面关掉了,忘记截图,下次一定!!!!! |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |