ubuntu16.04安装tensorflow-gpu以及pytorch教程(完整教程)

        实验室来了新电脑,重新搭框架。于是想整理一下教程,涉及搭框架的所有过程,主要是把别人的方式汇总。一步搞定所有过程!!!

1.安装linux系统

教程链接:https://blog.csdn.net/flyyufenfei/article/details/79187656

2.安装显卡驱动:

教程链接:https://www.linuxidc.com/Linux/2016-12/138907.htm

3.安装CUDA以及cudnn:

教程链接:https://blog.csdn.net/wanzhen4330/article/details/81699769

4.安装Anaconda(用于管理各种不同的深度学习环境,如tensorflow一个环境,pytorch一个环境)

4.1运行以下代码下载anaconda

 wget –c https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh

注:下载链接(到清华源找到5.2版本点击右键复制链接地址即可)安装5.2版本。

4.2安装

 sh Anaconda3-5.3.0-Linux-x86_64.sh     

5.安装tensorflow

5.1先创建tensorflow环境,python3.6版本不会报错

conda create –n tensorflow python=3.6

5.2激活环境               

conda activate tensorflow

5.3使用清华源安装tensorflow-gpu(这样安装的快)

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu

5.4检验是否安装成功

python 

 import tensorflow  (以后每次都需要先进去tensorflow环境,如果没有报错,就安装成功了)

6.安装pytorch

6.1创建pytorch环境并激活环境

conda create –n pytorch

conda activate pytorch

6.2安装pytorch

conda install pytorc torchvision -c pytorch

因为这会从国外下载安装,可能会很慢,如果安装的很慢,报http错误,

可以多试几次或者

建议使用国内源安装,教程如下:

教程链接:https://blog.csdn.net/yucicheung/article/details/79094657

6.3检验是否安装成功

python 

 import torch  (以后每次都需要先进去pytorch环境,如果没有报错,就安装成功了)

7.在jupyter notebook中使用tensorflow

Anaconda自带jupyter notebook,但是tensorflow安装在tensorflow环境中,所以无法在jupyter notebook里正常使用,需要额外安装ipython和jupyter包,安装成功后,通过 conda list命令可以看到jupyter和ipython。

教程如下:https://blog.csdn.net/xue_wenyuan/article/details/51545845

8.在jupyter notebook中使用pytorch

教程如下:https://www.jianshu.com/p/da89fb397ef6


问题:

双系统中Ubuntu16.04启动页面是黑屏,并只有一个光标在闪:

进去win10打开easyBCD删除原来的条目,重新建一个。

linux在用户登陆界面出现不断循环解决方法:

https://blog.csdn.net/ssmixi/article/details/73483795

猜你喜欢

转载自blog.csdn.net/yang_daxia/article/details/86549945
今日推荐