国内安装tensorflow 和 pytorch 具体流程(踩坑)

由于国内访问官方的镜像源不稳定,极容易产生连接中断,下载速度很忙的情况,导致只是单纯的pip install 和conda install 极其容易失败, 因为read time out(连接时间过长)产生错误终止,就算延长时间也是杯水车薪,没有很大用处。 以下是我近几日安装成功的流程:

Tensorflow

Tensorflow 最新版本和python 3.7 (anconda 的默认版本是3.7)不兼容, 当你import的时候就会报错了,所以这里推荐创建虚拟环境 python 3.6 或者 3.5, 我选择的是3.5. 然后激活虚拟环境, 再使用豆瓣的镜像源安装 tensorflow(试了几种,还是豆瓣的镜像源安装tensorflow最快最稳定)。
详细过程已有一篇博客详细描述了, https://blog.csdn.net/weixin_44319196/article/details/105112915 可以参阅其详细步骤。

Pytorch

1.首先检测自己的电脑是否有navida显卡, 是否支持cuda, 如何可以的话就能安装gpu版本了,也是pytorch的一个核心优势。 而如何检测电脑cuda 版本, 首先打开navida控制面板, 然后
国内安装tensorflow 和 pytorch 具体流程(踩坑)_第1张图片
然后点击系统信息。
国内安装tensorflow 和 pytorch 具体流程(踩坑)_第2张图片
NVCUDA.DLL 后的产品名称的数字就算版本号了。
然后通过官网安装cudatoolkit, 网址https://developer.nvidia.com/cuda-toolkit-archive, 这里建议安装低一个版本的,所以我安装了9.2. 安装过程很简单,就像安装正常软件一样, 安装成功后可通过 在cmd命令行里输入nvcc --version 来检测。
国内安装tensorflow 和 pytorch 具体流程(踩坑)_第3张图片
2. 接下来就是具体安装pytroch的流程了, 就算通过镜像源下载都会很慢, 所以推荐使用离线下载, 直接从国内镜像源内下载相应文件, 再离线安装。
去清华镜像源本地下载
(linux版本)https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/?C=M&O=A或https://anaconda.org/pytorch/pytorch/files?version=1.3.0
(windows版)https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/?C=M&O=A

pytorch-1.5.1-py3.5_cuda92_cudnn7_0.tar.bz2, 这是我下载的文件,1.51是pytorch最新版本, cuda92 是cuda版本, 3.5是你该环境的python版本。根据自身实际情况进行下载。

离线安装代码:conda install --offline 文件名/pytorch-1.5.1-py3.5_cuda92_cudnn7_0.tar.bz2 (该文件在电脑的绝对地址)

同样下载网址最下方的torchvision也要单独安装:conda install --offline 文件名/torchvision-0.6.1-py35_cu92.tar.bz2(该文件在电脑的绝对地址)

离线安装好后,还是需要重新利用conda来在线安装一次,要用它来对相关的包来进行在线匹配(也就是根据安装好的pytorch来匹配那些对应的好多库,不是说再下载一遍pytorch)。但是一定要注意了!!!在利用官网https://pytorch.org/代码时conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev一定要删掉后面的-c pytorch,因为它代表直接从官网去下载,不会与本地安装好的匹配,还会全部重新下载。所以改成:conda install pytorch torchvision cudatoolkit=9.2 -c defaults -c numba/label/dev它就会自动匹配之前离线安装好的东西,然后匹配相关的库的等级、版本之类的组件。
国内安装tensorflow 和 pytorch 具体流程(踩坑)_第4张图片
离线方案参照的是https://blog.csdn.net/qq_41936559/article/details/102699082?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-1.nonecase的博客,大家没那么清晰的话可以看看她的离线流程。
等安装完后, pytorch就可以使用了。 再在此虚拟环境下启动python IDE 就可以了。

最早安装这些东西在国外,很轻松毫无难度,然后回国后因为环境出问题了,就卸载了anaconda,重新在国内安装相应的库,才发现是真麻烦。 捣鼓了好几天,找到了我使用起来没问题的方法,希望能帮助到一些想要入坑的小白。

你可能感兴趣的:(国内安装tensorflow 和 pytorch 具体流程(踩坑))