Torchvision error: Could not find module image.pyd

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.