DefTruth

Results 214 comments of DefTruth

暂时没有测试过metal哈

use the latest nightly build on 20240110 solve this problem. ```bash python -m pip install --pre torch==2.3.0.dev20240110+cu121 torch-tensorrt==2.3.0.dev20240110+cu121 --extra-index-url https://download.pytorch.org/whl/nightly/cu121 python -m pip install --pre torchvision==0.18.0.dev20240110+cu121 --extra-index-url https://download.pytorch.org/whl/nightly/cu121 python -m...

but the nightly build after 20240110 is not avaliable ```bash ERROR: Could not find a version that satisfies the requirement torch-tensorrt==2.3.0.dev20240222+cu121 (from versions: 0.0.0.post1, 1.3.0, 1.4.0, 2.2.0.dev20231010+cu121, 2.2.0.dev20231106+cu121, 2.2.0.dev20231107+cu121, 2.2.0.dev20231108+cu121,...

```bash ERROR: Could not find a version that satisfies the requirement torch-tensorrt==2.3.0.dev20240222+cu121 (from versions: 0.0.0.post1, 1.3.0, 1.4.0, 2.2.0.dev20231010+cu121, 2.2.0.dev20231106+cu121, 2.2.0.dev20231107+cu121, 2.2.0.dev20231108+cu121, 2.2.0.dev20231109+cu121, 2.2.0.dev20231110+cu121, 2.2.0.dev20231111+cu121, 2.2.0.dev20231112+cu121, 2.2.0.dev20231113+cu121, 2.2.0.dev20231114+cu121, 2.2.0.dev20231115+cu121, 2.2.0.dev20231116+cu121, 2.2.0.dev20231117+cu121,...

@david-PHR hi ~ would you like to share your nsys profile results? I have encounter a similar problem that 'Myelin fused Attn but not run at MHA Kernel', see: -...

安装最新版本试试,最新的代码不会有这个log

可以提供下您的测试代码吗?

![image](https://github.com/DefTruth/statistic-learning-R-note/assets/31974251/07297b6a-d94c-4db0-aaef-8132071c94cb)