Tensorflow Pytorch安装

深度学习框架安装

由于深度学习框架包括Tensorflow,Pytorch 的源都是在国外的,在国内的安装有时候运气好的情况下可以下载成功,但是大部分的情况都是会出现超时的时候,下面介绍如何快速地安装Tensorflow,Pytorch框架。

Tensorflow 安装

  1. 安装Conda,并且使用Conda国内镜像源
  2. 创建虚拟环境
  3. 在虚拟环境中使用Conda或者Pip 下载Tensorflow

安装Conda

具体可以参考上两篇博文:

创建虚拟环境

conda create -n envs_name python=3.6 cudatoolkit=10.0 cudnn   #创建对应python3.6 cuda cudnn对应版本的环境
conda activate envs_name #激活虚拟环境

在虚拟环境中使用Conda 下载Tensorflow

对应GPU版本的tensorflow,需要安装相对应的显卡驱动,CUDA,Cudnn 库

conda install tensorflow=1.15 #安装CPU1.15版本的tensorflow
conda install tensorflow-gpu=1.15 #安装gpu版本的tensorflow

pip 下载tensorflow 此时也建议采用pip的镜像网站下载

pip install tensorflow==1.15 #安装CPU1.15版本的tensorflow
pip install tensorflow-gpu==1.15 #安装gpu版本的tensorflow

下载pytorch

方式一 conda 下载

国内环境下下载pytorch也是运气好的话,不会断,能成功,不好的话就GG的情况,所以依旧推荐使用清华镜像下载pytorch。步骤如下:

  1. 进入pytorch官网 (学习不训练的话采用CPU的版本,如果使用对GPU的话可以选择下面对应CUDA的版本),复制 conda install pytorch torchvision cpuonly 后面-c pytorch可以不用复制。

在这里插入图片描述

  1. 添加清华镜像
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

添加后直接在windows的终端输入上式复制的命令即可。

conda install pytorch torchvision cpuonly

或者使用pip安装

 pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -i https://pypi.tuna.tsinghua.edu.cn/simple

方式二 whl文件形式下载

进入清华源的网站,https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
windows是/win-64/ linux则是/linux-64/
选择合适的版本例如选用packname=1.3.1-cpu_py37XXXXX.tar.bz2 py_37为对应的python3.7版本
在这里插入图片描述
下载完成后,到对应的目录下 shift+鼠标左键打开power shell 输入下列指令即可

conda install packname  #packname即文件名
conda install torchvision

opencv 安装

pip install opencv-python  -i https://pypi.tuna.tsinghua.edu.cn/simple

安装错误集合(更新):

一 tensorflow

1.1.tensorflow遇到这种情况,则是

ImportError: No module named ‘_pywrap_tensorflow_internal’

都是因为这个C++的库没下载,可能由于某种愿意丢失了,通过加载安装该库即可
地址:https://aka.ms/vs/16/release/vc_redist.x64.exe 下载后依旧不能加载的话,重启电脑即可。

二 pytorch

2.1

Microsoft Visual C++ Redistributable is not installed, this may lead to the DLL load failure

都是因为这个C++的库没下载,可能由于某种愿意丢失了,通过加载安装该库即可
地址:https://aka.ms/vs/16/release/vc_redist.x64.exe 下载后依旧不能加载的话,重启电脑即可。

2.2

Error loading “D:\ProgramData\Anaconda3\envs\p38\lib\site-packages\torch\lib\asmjit.dll” or one of its dependencies.
  • 如上述,下载C++的redist库
  • 升级numpy库,卸载重装
  • 升级pytorch版本,确认版本与python cuda的匹配问题
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值