东方耀AI技术分享

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 活动 交友 discuz
查看: 3004|回复: 1
打印 上一主题 下一主题

[学习笔记] 002、【详细步骤】conda方式安装tensorflow-gpu==1.14和PyTorch==1.3

[复制链接]

1365

主题

1856

帖子

1万

积分

管理员

Rank: 10Rank: 10Rank: 10

积分
14437
QQ
跳转到指定楼层
楼主
发表于 2019-10-15 21:55:32 | 只看该作者 |只看大图 回帖奖励 |倒序浏览 |阅读模式
002、【详细步骤】conda方式安装tensorflow-gpu==1.14和PyTorch==1.3


conda方式安装tensorflow-gpu==1.14和PyTorch==1.3
前提条件:本机系统已经安装了CUDA:v10.0和cuDNN:v7.6.1 并不想改变 因为其他项目的TensorFlow-GPU使用了CUDA:v10.0这个版本
而查看PyTorch官网PyTorch==1.3的Stable版本只支持 CUDA v9.2和10.1版本 为了两者能共存需要用到conda的虚拟环境隔离


步骤:
1、conda create -n py3_torch_gpu python=3.6.8
2、conda activate py3_torch_gpu   conda list 查看pip版本
3、用pip --version  pip3 --version验证是否报错 以及位置是否在虚拟环境内部
4、pip 9.0.1 from D:\ProgramData\Miniconda3\envs\py3_torch_gpu\lib\site-packages (python 3.6)
5、执行:python get-pip.py 升级
6、再用pip --version pip3 --version 查看是否升级 同时根据路径只使用虚拟环境内的pip或pip3命令
通过 pip list 或 pip3 list也可以看出不同
7、conda list 查看pip版本 同时发现 Build=pypi_0  Channel=pypi  非常好!
8、pip3 install numpy和jupyter notebook和matplotlib和scikit-learn和nltk
9、【正式安装Pytorch】conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 发现pytorch torchvision非常卡 其他包很快(用的清华源) 失败!
10、conda install cudatoolkit=10.1  之后改变了环境内的pip版本 又得执行:python get-pip.py 升级
11、pip3 install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.html 速度非常快
12、检查PyTorch是否使用GPU(成功):import torch as T  T.__version__  T.cuda.current_device()   T.cuda.device_count()  T.cuda.get_device_name(0)
13、【安装tensorflow-gpu】pip3 install tensorflow-gpu==1.14  Successfully opened dynamic library nvcuda.dll
14、pip3 install keras    pip3 install opencv-python   pip3 install opencv-contrib-python




conda install tensorflow-gpu==1.14  会自动提示:
cudatoolkit        anaconda/pkgs/main/win-64::cudatoolkit-10.0.130-0
  cudnn              anaconda/pkgs/main/win-64::cudnn-7.6.0-cuda10.0_0
而PyTorch==1.3的Stable版本只支持 CUDA v9.2和10.1版本

2020年02-16:

  1. ==> WARNING: A newer version of conda exists. <==
  2. current version: 4.6.14
  3. latest version: 4.8.2
  4. Please update conda by running
  5. $ conda update -n base -c defaults conda
复制代码

【正式安装Pytorch=1.4】conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 发现 无法连接:CondaHTTPError: HTTP 000 CONNECTION FAILED for url
conda install cudatoolkit=10.1 共456.2M 下载速度比较快  之后改变了环境内的pip版本 又得执行:python get-pip.py 升级
pip3 install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html  速度非常快哦
但是 安装pytorch=1.4报错:pip\_internal\cli\base_command.py   MemoryError
解决:直接迅雷下载whl文件(会显示的根据你的版本):torch-1.4.0-cp36-cp36m-win_amd64.whl(这是我的 759M) 之后pip3 安装
再次运行pip3 install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
提示torch已经安装好了 后面再安装torchvision===0.5.0 pillow等 非常顺利  成功!
检查PyTorch是否使用GPU(成功)

【安装tensorflow-gpu】pip3 install tensorflow-gpu==1.15  Successfully opened dynamic library cudart64_100.dll
pip3 install keras


【正式安装Pytorch=1.5】conda install pytorch torchvision cudatoolkit=10.2 -c pytorch   成功啦(2020-05-08 有截图)






ipynb文件在附件,可提供下载!


东方老师AI官网:http://www.ai111.vip
有任何问题可联系东方老师微信:dfy_88888
【微信二维码图片】











00.png (117.29 KB, 下载次数: 160)

00.png

01.png (28.45 KB, 下载次数: 160)

01.png

02.png (46.19 KB, 下载次数: 157)

02.png

03.png (27.52 KB, 下载次数: 156)

03.png

04.png (27.43 KB, 下载次数: 158)

04.png

05.png (100.55 KB, 下载次数: 158)

05.png

06.png (45.48 KB, 下载次数: 157)

06.png

07.png (48.32 KB, 下载次数: 158)

07.png

08.png (39.19 KB, 下载次数: 156)

08.png

09.png (39.82 KB, 下载次数: 157)

09.png

000.jpg (73.42 KB, 下载次数: 157)

000.jpg

01.png (43.2 KB, 下载次数: 155)

01.png

torch===1.4.jpg (76.83 KB, 下载次数: 157)

torch===1.4.jpg

torch01.png (130.43 KB, 下载次数: 156)

torch01.png

torch02.png (116.04 KB, 下载次数: 155)

torch02.png

torch03.png (25.59 KB, 下载次数: 157)

torch03.png

Demo.ipynb

2.24 KB, 下载次数: 357

让天下人人学会人工智能!人工智能的前景一片大好!
回复

使用道具 举报

1365

主题

1856

帖子

1万

积分

管理员

Rank: 10Rank: 10Rank: 10

积分
14437
QQ
沙发
 楼主| 发表于 2020-6-23 14:37:15 | 只看该作者
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
让天下人人学会人工智能!人工智能的前景一片大好!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|人工智能工程师的摇篮 ( 湘ICP备2020019608号-1 )

GMT+8, 2024-5-4 03:24 , Processed in 0.192852 second(s), 22 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表