如果你正在面临配置环境的痛苦不妨进来看看

入门深度学习 , 很多人经历了从入门到放弃的心酸历程 , 且千军万马倒在了入门第一道关卡:环境配置问题 。俗话说 , 环境配不对 , 学习两行泪 。
如果你正在面临配置环境的痛苦 , 不管你是Windows用户、Ubuntu用户还是苹果死忠粉 , 这篇文章都是为你量身定制的 。接下来就依次讲下Windows、Mac和Ubuntu的深度学习环境配置问题 。
一、Windows系统深度学习环境配置
系统:Win1064位操作系统
安装组合:Anaconda+PyTorch(GPU版)+GTX1060
开源贡献:伍天舟 , 内蒙古农业大学
1.1 打开Anaconda Prompt
1、conda create -n pytorch python=3.7.0:创建名为pytorch的虚拟环境 , 并为该环境安装python=3.7 。2、activate pytorch:激活名为pytorch的环境

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
因为官方源太慢了 , 这里使用清华源下载
1.4 镜像中下载对应的安装包
清华镜像:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
pytorch:

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
【解决方法】: https://www.cnblogs.com/hellojiaojiao/p/10790273.html
conda下载太慢问题
【解决方案】https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
我直呼一声清华NB!
cuda安装
cuda历史版本下载:https://developer.nvidia.com/cuda-toolkit-archive 要看NVIDIA的组件 , 自己的CUDA支持哪个版本 , 我是1060显卡 , 所以我下的10.0版本的cuda

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
pkg 安装包后点击进入 , 按下一步完成安装即可 。默认安装地点为~/opt 。想用command line install 的 , 请自行参考:https://docs.anaconda.com/anaconda/install/mac-os/
2.2 确认下载情况

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
pytorch 官网显示当前设备应该用的下载语句 。此情况 , 我的是
conda install pytorch torchvision -c pytorch
复制下来 , 粘贴到Terminal 中运行就可以开始下载了 。
【如果你正在面临配置环境的痛苦不妨进来看看】 下载完成后 , 在Terminal 输入
python3
之后import 两个刚下载的包 , 确认下载完成
import torchimport torchvision print(torch.__version__)print(torchvision.__version__)如果import 和输出正常 , 配置就完成了!输入

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
下载完后的文件名是:Anaconda3-2020.02-Linux-x86_64.sh 。
cd到Anaconda3-2020.02-Linux-x86_64.sh所在的目录:

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
激活刚刚创建的虚拟环境:conda activate pytorch在安装之前先添加下面这个源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch然后输入下面代码:
conda install pytorch torchvision cpuonly -c pytorch

如果你正在面临配置环境的痛苦不妨进来看看

文章插图

如果你正在面临配置环境的痛苦不妨进来看看

文章插图
然后将cudnn解压后的include和lib64文件夹复制到cuda中
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include #解压后的文件夹名字为cuda-10.2 sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h /usr/local/cuda-10.2/lib64/libcudnn*
3.3.4 安装pytorch-gpu

如果你正在面临配置环境的痛苦不妨进来看看
文章插图
conda安装:
# 选择自己对应的cuda版本conda install pytorch torchvision cudatoolkit=10.2
pip安装:
pip install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple
四、写在最后
所有的深度学习环境安装指南到这里就结束了 , 希望能解决你面临的环境配置难题 。关于实践项目 , 可以结合阿里天池的学习赛进行动手实践 。数据挖掘学习赛(进行中 , 5832人参与) https://tianchi.aliyun.com/competition/entrance/231784/forum cv实践学习赛(进行中 , 1933参与) https://tianchi.aliyun.com/competition/entrance/531795/forum nlp实践学习赛(进行中 , 573人参与) https://tianchi.aliyun.com/competition/entrance/531810/forum
责任编辑:xj
原文标题:深度学习环境配置指南!(Windows、Mac、Ubuntu全讲解)
文章出处:【微信公众号:人工智能与大数据技术】欢迎添加关注!文章转载请注明出处 。

    推荐阅读