热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

[linux]Ubuntu18.04+CUDA+CUDNN+Pytorch[总结]

1.序言由于下载了Github的代码,代码运行需要该配置环境RequiredenvironmentUbuntu16.04orabovePyTorch0.4.1oraboveCUDA

1. 序言

由于下载了Github的代码,代码运行需要该配置环境


Required environment

  • Ubuntu 16.04 or above
  • PyTorch 0.4.1 or above
  • CUDA 9.1 or above

但是本机安装的18.04 版本的ubuntu, 所以自己独立开始配置, 在配置过程中,出现了很多问题.

如:1.  Nvidia 显卡的算力是 8.6 , 但是安装不同版本的pytorch-cuda 所允许的算例不一样

2. 安装的 pytorch cuda cudnn等版本不匹配

3. 使用了pytorch官网的conda命令行来安装,但是安装的结果显示是cpu版本的pytorch.所以建议使用pip,为了避免错误和麻烦

4. 建议按着该教程以下的步骤,一步步开始 


二 查看电脑显卡的算力 


推荐阅读
author-avatar
x75066882
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有