换源问题
打算跑一下yolov10,我看的教程在安装pytorch的时候强调了不要换源直接下(可能导致调用不了GPU)。但是我直接pip默认用了清华源,所以恢复了一下默认源。
版本问题
我自己的电脑是cuda11.7,去pytorch官网找对应的下载代码是这样
但是下载成功之后执行torch.__version__显示是cpu,也就是根本调用不了gpu。
PS:用conda下载的时候Solving Environment一直转圈
解决方法
pip3 install torch==2.0.1+cu117 torchvision==0.15.2+cu117 torchaudio==2.0.2+cu117 -f https://download.pytorch.org/whl/cu117/torch_stable.html
安装成功!