我需要torch和torchvision与GPU支持。Torch和Torchaudio安装为GPU版本，但Torchvision安装为CPU版本
> conda list
...
torch 2.0.0+cu117 pypi_0 pypi
torchaudio 2.0.1+cu117 pypi_0 pypi
torchvision 0.13.1 cpu_py310h378ed51_0
...此Python脚本显示CUDA可用
from torch import cuda
print(cuda.is_available()) # True我有Windows11，CUDA 11。7、Anaconda如何安装支持GPU的torchvision？
1条答案
按热度按时间ljo96ir51#
这个解决办法对我很有效
1.备份Conda环境
1.卸载并重新安装Anaconda
1.运行
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117