深度学习框架安装
由于深度学习框架包括Tensorflow,Pytorch 的源都是在国外的,在国内的安装有时候运气好的情况下可以下载成功,但是大部分的情况都是会出现超时的时候,下面介绍如何快速地安装Tensorflow,Pytorch框架。
Tensorflow 安装
- 安装Conda,并且使用Conda国内镜像源
- 创建虚拟环境
- 在虚拟环境中使用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。步骤如下:
- 进入pytorch官网 (学习不训练的话采用CPU的版本,如果使用对GPU的话可以选择下面对应CUDA的版本),复制 conda install pytorch torchvision cpuonly 后面-c pytorch可以不用复制。
- 添加清华镜像
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的匹配问题