1.我的设备中这样安装,先安装CUDA_11.6.0_511.23_windows.exe,再安装cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive(这个是靠移动的),在pip install pytorch和torchvision的whl文件。
2.如果需要安装paddle的,要提前看看paddle官网的安装教程,里面十分详细:
3.CUDA安装前需要安装vs2019,遇到CUDA安装错误,再来一次,哪个安装不了的就取消勾选哪一个
4.需要提前判断本机可安装哪个版本的CUDA、torch和torchvison&#