Hello,
after fighting to install for GPU pytorch with anaconda on linux ubuntu, I have a new problem how to download torchvision for GPU any recommendations?
I have installed it and run yolo v7 successfully with pip, but I don’t want to work with pip, because in my anaconda path I have the rapids of NVIDIA.
thank you in advance
The install instructions from here work for me:
conda install pytorch torchvision pytorch-cuda=11.7 -c pytorch -c nvidia
...
>>> import torch
>>> import torchvision
>>> torch.__version__
'1.13.1'
>>> torchvision.__version__
'0.14.1'
>>> torch.version.cuda
'11.7'
>>> torchvision.version.cuda
11070