Pytorch 编译cpp、cuda扩展遇到的问题

参考：https://blog.csdn.net/tanmx219/article/details/100829920

问题一：

C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\cpp_extension.py:287: UserWarning: Error checking compiler version for cl: 'utf-8' codec can't decode byte 0xd3 in position 0: invalid continuation bytewarnings.warn('Error checking compiler version for {}: {}'.format(compiler, error))

utf-8的问题，找到报错目录cpp_extension.py的对应代码位置：

也就是283行re.search这里出了问题，把原来的：

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode().strip())

改成

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode(' gbk').strip())

究其原因可能是visual studio的那个cl，原来输出是英文的，直接utf-8解码没问题，新版更加“人性化”的能输出中文，然后pytorch这里没改还是utf-8…所以让pytorch按gbk解码就好

问题二：

Traceback (most recent call last):File "C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\cpp_extension.py", line 1539, in _run_ninja_buildenv=env)File "C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\subprocess.py", line 512, in runoutput=stdout, stderr=stderr)
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

还是这个cpp_extension.py，找到对应位置：

也就是这个_run_ninja_build方法，把下面这个

command = ['ninja', '-v']

改成

command = ['ninja','--v']

或者

command = ['ninja','--version']

好像只有一种改法有效，大概率为后者。这个东西的作用其实单纯就是指定build.ninja文件的第一行输出：

Pytorch 编译cpp、cuda扩展遇到的问题相关推荐

如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应
转载请注明: 仰望高端玩家的小清新 http://www.cnblogs.com/luruiyuan/ 如何解决pytorch 编译时CUDA版本与运行时CUDA版本不对应如果pytorch的编译时 ...
Pytorch 编译cpp、cuda扩展时卡在import
具体表现就是import的时候不动了: 其中的Encoder就是需要cpp.cuda编译扩展的部分此时如果停止运行,可以看到如下提示(jupyter notebook能看到,pycharm好像看不到 ...
详解PyTorch编译并调用自定义CUDA算子的三种方式
点击上方"3D视觉工坊",选择"星标" 干货第一时间送达在上一篇教程中,我们实现了一个自定义的CUDA算子add2,用来实现两个Tensor的相加.然后用Py ...
PyTorch中的C++扩展
今天要聊聊用 PyTorch 进行 C++ 扩展. 在正式开始前,我们需要了解 PyTorch 如何自定义module.这其中,最常见的就是在 python 中继承torch.nn.Module,用 ...
自定义 C++ 和 CUDA 扩展
来源官方文档前言 PyTorch 提供了大量与神经网络.随机张量代数(arbitrary tensor algebra).数据整合(data wrangling)以及其他目的相关的操作.但是,您仍 ...
php pthread安装编译,php 多线程扩展 pthreads 安装及使用
1.扩展的编译安装php(Linux),编辑参数 --enable-maintainer-zts 是必选项: 2.下载 php7: http://tw2.php.net/get/php-7.1.2.t ...
matlab怎么跑.cpp程序,MATLAB编译cpp文件
目的:打通MATLAB与VC之间的通道,实现用MATLAB调用VC程序,以及VC调用MATLAB程序. 上篇博客实现了将MATLAB的M文件编译成C/C++文件,exe应用程序.这篇实现MATLAB编 ...
编译安装 PHP 扩展 - 含 swoole 扩展安装和升级
编译安装 Redis 扩展 - phpredis 编译安装 Swoole 扩展编译安装 Pcntl 扩展编译安装 Sockets 扩展编译安装 Event 扩展编译安装 Mongodb 扩展 ...
[CentOS]CentOS下编译CPP文件时报错[undefined reference to `__gxx_personality_v0' collect2: ld]的解决办法...
在CentOS环境下编译CPP时报出 undefined reference to `__gxx_personality_v0' collect2: ld 以上错误,调查了一下,加上参数[-lstdc ...

Pytorch 编译cpp、cuda扩展遇到的问题

Pytorch 编译cpp、cuda扩展遇到的问题相关推荐

最新文章

热门文章