Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor' #765

Open
Haremakhet opened this issue Mar 14, 2024 · 7 comments

Comments

@Haremakhet
Copy link

Traceback (most recent call last):
File "/content/Real-ESRGAN/inference_realesrgan.py", line 5, in
from basicsr.archs.rrdbnet_arch import RRDBNet
File "/usr/local/lib/python3.10/dist-packages/basicsr/init.py", line 4, in
from .data import *
File "/usr/local/lib/python3.10/dist-packages/basicsr/data/init.py", line 22, in
_dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
File "/usr/local/lib/python3.10/dist-packages/basicsr/data/init.py", line 22, in
_dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
File "/usr/lib/python3.10/importlib/init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "/usr/local/lib/python3.10/dist-packages/basicsr/data/realesrgan_dataset.py", line 11, in
from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels
File "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py", line 8, in
from torchvision.transforms.functional_tensor import rgb_to_grayscale
ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

@RohitPoul
Copy link

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py"
do this change in the code - torchvision.transforms.functional import rgb_to_grayscale
you have to replace it every single use

@tushar-31093
Copy link

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py" do this change in the code - torchvision.transforms.functional import rgb_to_grayscale you have to replace it every single use

So does this imply that this repo wont function anymore without explicity changing this? At the same time colab wont work either?

@viniciusmods
Copy link

`!nvidia-smi

from PIL import Image
import cv2, os, subprocess
from google.colab import drive
from tqdm import tqdm

!git clone https://github.com/xinntao/Real-ESRGAN.git
%cd Real-ESRGAN

with open('requirements.txt', 'w') as arquivo:
arquivo.write("basicsr==1.3.5\ngfpgan>=1.3.5\nnumpy\nopencv-python\nPillow\ntorch>=1.7\ntorchvision\ntqdm\n")

!pip install facexlib ffmpeg ffmpeg-python
!pip install -r requirements.txt
!python setup.py develop

with open('/usr/local/lib/python3.10/dist-packages/basicsr-1.3.5-py3.10.egg/basicsr/data/degradations.py', 'r') as file:
file_data = file.read()
file_data = file_data.replace('from torchvision.transforms.functional_tensor import rgb_to_grayscale', 'from torchvision.transforms.functional import rgb_to_grayscale')
with open('/usr/local/lib/python3.10/dist-packages/basicsr-1.3.5-py3.10.egg/basicsr/data/degradations.py', 'w') as file:
file.write(file_data)

mount_drive=False #@param{type:"boolean"}

if mount_drive:
drive.mount('/content/gdrive/')`

@JossCamp
Copy link

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

@rahulmahmud
Copy link

go to the file show here "/usr/local/lib/python3.10/dist-packages/basicsr/data/degradations.py" do this change in the code - torchvision.transforms.functional import rgb_to_grayscale you have to replace it every single use

please give a proper video

@eharsh2711
Copy link

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

Can confirm, this worked for me in colab.

@m4ra7h0n
Copy link

m4ra7h0n commented Apr 4, 2024

I don't know why in Colab, whenever something gets updated, it always has to mess up everything else. To solve this problem, you need to execute this command.

!python -m pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118

After this, everything should work fine.

Can confirm, this worked for me in colab.

ERROR: Could not find a version that satisfies the requirement torch==2.0.1 (from versions: 2.2.0, 2.2.1, 2.2.2)
ERROR: No matching distribution found for torch==2.0.1

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

8 participants