Torchvision error: Could not find module image.pyd
more_vert
Error context
I installed PyTorch in Anaconda with CUDA 10.2 as compute platform via the following command line:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
The following error shows up:
UserWarning: Failed to load image Python extension: Could not find module 'D:\conda\envs\py38\Lib\site-packages\torchvision\image.pyd' (or one of its dependencies). Try using the full path with constructor syntax.
warn(f"Failed to load image Python extension: {e}")
Fix this issue
There are two approaches to fix these two issues based on my research:
Approach 1 - use CUDA 11.3
Use the following command to use CUDA 11.3 as compute platform:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
Approach 2 - rename image.pyd to image.py
I have not verified this approach but found it from internet (Chinese).
*.pyd files are Python extension modules compiled from other languages like C.
This approach is to manually rename 'image.pyd' in torchvision package folder (for example, the path is 'D:\conda\envs\py38\Lib\site-packages\torchvision' in my computer) to 'image.py'.
Let me know if you can get it work.
info Last modified by Raymond 2 years ago
copyright
This page is subject to Site terms.
Log in with external accounts
warning Please login first to view stats information.
article
Implement SCD Type 2 Full Merge via Spark Data Frames
article
Apache Kafka Python Producer and Consumer Clients Introduction
code
Python - Read Messages in Kafka Topic
article
Convert String to Date using Python
code
Pandas DataFrame Group by one Column and Aggregate using MAX, MIN, MEAN and MEDIAN
Read more (92)