Torchvision error: Could not find module image.pyd

visibility 600 access_time 23d languageEnglish

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 23d copyright This page is subject to Site terms.

Please log in or register to comment.

account_circle Log in person_add Register

Log in with external accounts

timeline Stats
Page index 32.96
local_offer Tags