[环境搭建十一] 安装tensorflow2.0-gpu，torch-1.4.0-白红宇的个人博客

发布日期：2021-05-07 00:09:17 浏览次数：22 分类：精选文章

本文共 329 字，大约阅读时间需要 1 分钟。

注意事项：

（1）tensorflow最好使用conda安装:

conda install tensorflow-gpu=2.0.0

conda可以帮忙协调cuda和cudnn，自己用pip安装维护太麻烦了。

（2）tensorflow-gpu与cuda，cudnn版本的对应关系

版本	cuDNN、CUDA
tensorflow-2.1.0	7.6、10.1
tensorflow-2.0.0	7.4、10.0

（3）安装pytorch

pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

上一篇：[环境搭建十二] ubuntu配置命令

关于作者

喝酒易醉，品茶养心，人生如梦，品茶悟道，何以解忧？唯有杜康！

-- 愿君每日到此一游！

Git 配置SSH公钥、私钥 2019-03-06

极客时间离线课堂 2019-03-06

Spring Session 2019-03-06

koa2 中间件里面的next到底是什么 2019-03-06

在create-react-app创建的项目下允许函数绑定运算符 2019-03-06

博客园新闻频道开始公开测试 2019-03-06

评论表聚集索引引起的评论超时问题 2019-03-06

博客园上海俱乐部4月份活动通知邀请函已经发出！ 2019-03-06

上周热点回顾（5.24-5.30） 2019-03-06

Internet Explorer 10 专题上线 2019-03-06

上周热点回顾（12.31-1.6） 2019-03-06

上周热点回顾（1.21-1.27） 2019-03-06

上周热点回顾（6.3-6.9） 2019-03-06

上周热点回顾（8.12-8.18） 2019-03-06

【故障公告】升级阿里云 RDS SQL Server 实例故障经过 2019-03-06

蹒跚来迟：新版博客后台上线公测 2019-03-06

上周热点回顾（9.16-9.22） 2019-03-06

上周热点回顾（11.4-11.10） 2019-03-06

白红宇的个人博客 - 记录点点滴滴的事 - 您是第 459522031 位访客