https://github.com/NVIDIA/apex/issues/605
FusedLayerNorm leads to RuntimeError: CUDA error: no kernel image is available for execution on the device · Issue #605 · NVID
After a GPU tensor goes through FusedLayerNorm, the next time that memory is accessed I get a RuntimeError: CUDA error: no kernel image is available for execution on the device. To reproduce: impor...
github.com
Recent changes to Pytorch's built-in extension builder sometimes lead to it compiling for a different architecture.
라고함.
export TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5"
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
'Python Library > Machine Learning' 카테고리의 다른 글
| [DNN] Day 04_shallow_neural-net_With_Vectorization (0) | 2019.07.28 |
|---|---|
| [DNN] Day 03_Logistic_Classification_With_Vectorized_Algorithm (0) | 2019.07.28 |
| [CNN] Day 03_vision03_Multiclass_Classification (0) | 2019.07.27 |
| [CNN] Day 03_vision02_Model_Transfer_With_Augmentation (0) | 2019.07.27 |
| [CNN] Day 02_vision01_basicOfCNN (0) | 2019.07.27 |