中国移动 二分类数据集 UEditor Anaconda ScrollView Zookeeper安装 远程桌面登陆 canal安装 excel xml node menu vue动态绑定class 后台模板 jquery循环 jquery解析json数据 jquery清除子元素 ajax里面可以嵌套ajax吗 mysql分区表优劣分析 手动安装fastboot驱动 mysqlinsert python编程题 python正则表达式例子 python建站 java编译 java的集合框架 java数组追加 java声明变量 python源码 冬青鼠 手机照片恢复免费软件 vbs表白代码 max电池容量 js日期格式化 我的世界透视 拼多多商家下载 pr时间轴 geartrax 拍照姿势的摆法女 lol修改皮肤
当前位置: 首页 > 学习教程  > python

【pytorch】——3080源码编译pytorch1.1.0

2021/2/6 22:30:39 文章标签: 测试文章如有侵权请发送至邮箱809451989@qq.com投诉后文章立即删除

版本:1.1.0 现在能编译通过,但是不能使用gpu 1. 依赖 3080cuda10.1(30系列只支持cuda11)python3.6.9 2. 流程 编译之前需要声明一个环境变量: export TORCH_CUDA_ARCH_LIST"7.5" 其他参考pytorch官方的说明&#x…

版本:1.1.0

现在能编译通过,但是不能使用gpu

1. 依赖

  • 3080
  • cuda10.1(30系列只支持cuda11)
  • python3.6.9

2. 流程

编译之前需要声明一个环境变量:
export TORCH_CUDA_ARCH_LIST="7.5"

其他参考pytorch官方的说明,pytorch

3. 可能问题

  1. nvcc fatal : Unsupported gpu architecture ‘compute_86‘
    声明前面的环境变量

4. python setup.py develop

可参考:

  • https://zhuanlan.zhihu.com/p/55889981
  • https://zhuanlan.zhihu.com/p/55204134
  1. 完成之后会在conda环境的site-package中torch.egg-link文件
    在这里插入图片描述

  2. torch.egg-link文件其实就是指向了源码编译的路径

cat torch.egg-link
>>>/home/darknet/CM/15_pytorch/pytorch110
  1. 在上面指向的路径下有torch文件夹,就是torch模块的路径。可以通过python要验证
>>> import torch
>>> torch.__file__
'/home/darknet/CM/15_pytorch/pytorch110/torch/__init__.py'
  1. 修改源码
  • 如果改变了一个pytroch的python源文件会自动被接收到,不需要运行任何命令使python的解释器看到这些改变。修改python文件不需要编译
  • 如果改变了扩展库的一个c++源文件,可以重新运行develop命令,它会重新构建这个扩展。修改c++源码需要重新编译

本文链接: http://www.dtmao.cc/news_show_2000139.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?