使用命令 pip uninstall torchvision pytorch torchaudio torch 之后用nvidia-smi查看对应cuda 版本 然后使用这个命令 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 后面的113换成你的cuda版本对应的(我的是11.3所以就是113) 这样下载下来的就不是cpu版本的了...
使用命令 pip uninstall torchvision pytorch torchaudio torch 之后用nvidia-smi查看对应cuda 版本 然后使用这个命令 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 后面的113换成你的cuda版本对应的(我的是11.3所以就是113) 这样下载下来的就不是cpu版本的了...
使用命令 pip uninstall torchvision pytorch torchaudio torch 之后用nvidia-smi查看对应cuda 版本 然后使用这个命令 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 后面的113换成你的cuda版本对应的(我的是11.3所以就是113) 这样下载下来的就不是cpu版本的了...
使用命令 pip uninstall torchvision pytorch torchaudio torch 之后用nvidia-smi查看对应cuda 版本 然后使用这个命令 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 后面的113换成你的cuda版本对应的(我的是11.3所以就是113) 这样下载下来的就不是cpu版本的了...