用Anaconda安装深度学习库全过程

目前,在Windows用Anaconda安装、管理Python的库已经非常非常方便(以前安装过非常多次,各种途径混用非常恶心),故本文总结全程用Anaconda安装深度学习库(全程不混用pip)。另外:

  • 本文使用的Anaconda自带的Python版本是3.7.9,一般3.7以上可以满足所有的库的要求;
  • 用nvidia-smi查看显卡驱动可得(记得提前把显卡驱动装好,否则cmd中没有该命令):本机的cuda版本是11.1,driver版本457.51,显卡是英伟达的2060s 8G。

正常安装完Anaconda后(记得勾选加入环境变量),按下面顺序安装即可:

(1)用conda创建一个虚拟环境:

  • conda create -n tf_torch python=3.7.9
  • conda activate tf_torch
  • (退出环境:conda deactivate)
  • (删除环境:conda remove -n tf_torch --all)
    说明:创建虚拟环境是好习惯,假如库安装出错就把这个环境直接删了再建即可。若用remove删不干净,就在Anaconda安装目录下的envs文件夹里把对应文件夹删了就行。

(2)深度学习库安装:

  • conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
  • conda install cudnn=7.6.5
  • conda install tensorflow-gpu=2.3.0
    说明:不论是pytorch还是tensorflow,都需要cuda和cudnn。本顺序的好处是:pytorch的官方安装命令会自动包括cuda10.1(这个版本号可以随便修改),然后只用再安装cudnn和tensorflow库即可。值得注意的是:TF2.3.0只和cuda10.1匹配,故不要搞最高级的!

(3)jupyter显示虚拟环境并按tab补全:在虚拟环境tf_torch中进行

  • conda install ipython —— 里面自带jedi库
  • conda install jedi=0.17.2 —— 升/降级jedi库到一个稳定版本
  • conda install ipykernel —— 里面自带jupyter-core库
    说明:Anaconda安装好是自带jupyter的,但是直接打开它是找不到虚拟环境。即使碰巧找到,按tab键也是无法自动补全代码的,这是因为该虚拟环境里要么是缺少jedi和jupyter-core库,要么是这两个库的版本没有匹配好导致的(目前是因为jedi版本太旧或太新都不行,jupyter-core可以用最新版本)!第(2)结束后会发现ipython还没安装,安装它会自动捆绑一个jedi库,记得查看它的版本号;然后安装ipykernel库,它会自动捆绑一个jupyter-core库。—— 注意:无需在虚拟环境下再安装一个jupyter

(4)安装其他必备库:

  • conda install matplotlib
  • conda install pandas
    说明:Numpy由于是TensorFlow库的依赖项,所以早被捆绑安装好了。

(5)我个人需要的3D绘图库Mayavi2:在虚拟环境tf_torch中进行

  • conda install mayavi
  • conda install -c conda-forge importlib_resources
  • conda install -c anaconda ipywidgets
  • conda install -c conda-forge ipyevents
    说明:目前的mayavi2仅适配python3.6以及全部3.7.x,不支持<3.6和>3.8版本的python(注意前面的tensorflow2.3.0适配>3.7的python!)。后3个全是mayavi2的依赖库,其中最后两个是为了mayavi2在jupyter环境下也能使用。仅安装ipywidgets和ipyevents还不够,还需下面的设置:
  • jupyter nbextension install --py mayavi --user
  • jupyter nbextension enable --py mayavi --user

总结:如今Anaconda管理Python包已经非常非常全面方便!因此本文全程用conda命令进行安装,没有使用pip,我也不推荐使用conda和pip混用!!另外,如果直接用conda install xxx安装某个库发现无法直接找到,那么可以用科学xx上网在Google搜一下:conda install xxx,就会在Anaconda Cloud中找到能正常下载的命令!例如mayavi2中后3个库都无法直接用conda install xxx安装,这些可用的命令都是在Anaconda Cloud中找到/推荐的。


测试:

  • Tensorflow-gpu的测试:tf.test.is_gpu_available()
  • PyTorch-gpu的测试:torch.cuda.is_available()
  • Mayavi2测试:from mayavi import mlab
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,172评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,346评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,788评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,299评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,409评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,467评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,476评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,262评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,699评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,994评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,167评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,827评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,499评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,149评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,387评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,028评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,055评论 2 352

推荐阅读更多精彩内容