TX2的配置
目录:
- PYCARM的安装
- archiconda的安装
- Pytorch和torchvision的安装
- scikit-image的安装
- DCNV2可变形卷积的安装
- yolact++的问题 “too many resources requested for launch”
前言:
这是TX2配置最后的章节,当运行yolact++代码的时候,会出现这样的error:
“too many resources requested for launch”
我当初一看这指定是这个机子算力不够啊。开始在TX2的论坛里面疯狂寻找。一开始找到的是pytorch下cuda线程数的原因,但是这个问题如果安装我说的通过TX2官方的方案安装pytorch就不会有问题。因为已经调整过线程数。找了一圈发现好像是DCNV2下cuda的线程数的问题。下面是解决方案。
yolact++的问题 “too many resources requested for launch”
这个主要是因为线程的问题,原本的DCNv2的1024线程数量小小的TX2没有办法承受,因此需要修改一下。
找到下面这两个文件:
src / cuda / dcn_v2_im2col_cuda.cu
src / cuda / dcn_v2_psroi_pooling_cuda.cu
更改下面对应的行:(将1024改为512,两个文件都要进行修改!!)
const int CUDA_NUM_THREADS = 1024;
更改为:
const int CUDA_NUM_THREADS = 512;