search

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
8.4k stars 1.4k forks source link

ModuleNotFoundError: No module named 'fused_layer_norm_cuda' #1593

Open zhuofalin opened 1 year ago

zhuofalin commented 1 year ago

Describe the Bug

ModuleNotFoundError: No module named 'fused_layer_norm_cuda'

Environment

import apex  # ok

but, raise error ModuleNotFoundError: No module named 'fused_layer_norm_cuda' , when i use in the project

**Expected Behavior**
poonehmousavi commented 1 year ago

I have he same issue with Cuda 11.7

yangjiangeyjg commented 1 year ago

I have he same issue with Cuda 11.7

Have you solved it yet？

yangjiangeyjg commented 1 year ago

Describe the Bug

ModuleNotFoundError: No module named 'fused_layer_norm_cuda'

Environment

  • 1 My OS is windows10, pytorch version is 1.13.1+cu116 and cuda is 11.6 Today, i find a problem: ModuleNotFoundError: No module named 'fused_layer_norm_cuda'
import apex  # ok

but, raise error ModuleNotFoundError: No module named 'fused_layer_norm_cuda' , when i use in the project

Expected Behavior

Have you solved it yet？

poonehmousavi commented 1 year ago

No I could not manage to solve yet

poonehmousavi commented 1 year ago

Here is the setting I have run mu experiments on: NVIDIA-SMI 510.47.03 Driver Version: 510.47.03 CUDA Version: 11.6