Allow io.write_video to work with video_arrays on different devices

[bmmtstb]

torchvision.io.write_video() fails if the video_array is on other devices, due to .numpy() not being able to convert properly.

Used .numpy(force=True) to make sure tensors on other devices get converted properly.

• Added tests (which I can't test locally due to PyAV)

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/8537

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 15 New Failures, 6 Unrelated Failures

As of commit bd6f0a4 with merge base 4a1cb63 ():

NEW FAILURES - The following jobs have failed:

• CMake / windows (windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• CMake / windows (windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Lint / python-types / linux-job (gh)
  RuntimeError: Command docker exec -t d08e198640bd6240c9805ac6d7b379fda6739f2c322cb539e0cbb4f318447bda /exec failed with exit code 1
• Tests / unittests-linux (3.10, linux.12xlarge, cpu) / linux-job (gh)
  test/test_io.py::TestVideo::test_write_video_with_audio[cuda]
• Tests / unittests-linux (3.8, linux.12xlarge, cpu) / linux-job (gh)
  test/test_io.py::TestVideo::test_write_video_with_audio[cuda]
• Tests / unittests-macos (3.10, macos-m1-stable) / macos-job (gh)
  test/test_io.py::TestVideo::test_write_video_with_audio[cuda]
• Tests / unittests-macos (3.12, macos-m1-stable) / macos-job (gh)
  test/test_io.py::TestVideo::test_write_video_with_audio[cuda]
• Tests / unittests-macos (3.8, macos-m1-stable) / macos-job (gh)
  The process '/usr/bin/git' failed with exit code 128
• Tests / unittests-macos (3.9, macos-m1-stable) / macos-job (gh)
  test/test_io.py::TestVideo::test_write_video_with_audio[cuda]
• Tests / unittests-windows (3.10, windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Tests / unittests-windows (3.11, windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Tests / unittests-windows (3.12, windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Tests / unittests-windows (3.8, windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Tests / unittests-windows (3.8, windows.g5.4xlarge.nvidia.gpu, cuda, 11.8) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128
• Tests / unittests-windows (3.9, windows.4xlarge, cpu) / windows-job (gh)
  The process 'C:\Program Files\Git\cmd\git.exe' failed with exit code 128

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

• Tests / unittests-linux (3.12, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
  The process '/usr/bin/git' failed with exit code 128
• Tests / unittests-linux (3.9, linux.12xlarge, cpu) / linux-job (gh) (matched linux rule in flaky-rules.json)
  The process '/usr/bin/git' failed with exit code 128

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Build Windows Conda / pytorch/vision / upload / conda-py3_8-cpu (gh) (trunk failure)
  Unable to find any artifacts for the associated workflow
• Build Windows Conda / pytorch/vision / upload / conda-py3_8-cuda11_8 (gh) (trunk failure)
  Unable to find any artifacts for the associated workflow
• Build Windows Conda / pytorch/vision / upload / conda-py3_8-cuda12_1 (gh) (trunk failure)
  Unable to find any artifacts for the associated workflow
• Build Windows Conda / pytorch/vision / upload / conda-py3_8-cuda12_4 (gh) (trunk failure)
  Unable to find any artifacts for the associated workflow

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[NicolasHug]

Thank you @bmmtstb