Pytorch(GPU)安装小结

您所在的位置:网站首页 pytorch作用 Pytorch(GPU)安装小结

Pytorch(GPU)安装小结

2023-01-28 18:59| 来源: 网络整理| 查看: 265

引:最近在学习神经网络的搭建与使用,需要安装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,找到这个库卸载掉就可以

Pytorch(GPU)安装小结_自动下载

conda list //conda查询安装库命令conda uninstall cpuonly //conda卸载命令pip uninstall cpuonly //python卸载命令(2)无cpuonly库

如果查询cpuonly并不存在,(骚操作)此时要先安装cpuonly库然后在卸载,这个过程中会重新安装一个适配的pytorch库(应该是GPU版的)

装完太高兴直接把页面关掉了,忘记截图,下次一定!!!!!



【本文地址】


今日新闻


推荐新闻


    CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3