pytorch1.13.1
torch-1.13.1+cu117-cp37torchvision-0.14.1+cu117-cp37
cuda_11.7.1_516.94_windows.exe
efficientnet_pytorch-0.6.3
segmentation_model.pytorch-0.3.2
下载链接:
【1】Start Locally | PyTorch
【2】https://download.pytorch.org/whl/torch_stable.html
【3】CUDA Toolkit:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
【4】nvidia驱动程序:https://www.nvidia.cn/Download/index.aspx?lang=cn
【5】cuDNN Archive:https://developer.nvidia.com/rdp/cudnn-archive
【6】【新提醒】Pytorch深度学习环境配置 - 深度学习 - Hello Mat - Powered by Discuz! (halcom.cn)
【1】device这里cuda:0,这里:不能有空格device = torch.device("cuda:" + str(GPU_index) if torch.cuda.is_available() else "cpu")【2】cannot import name ‘container_abcs‘ from ‘torch._six‘错误的解决方法
【3】libtorch-win-shared-with-deps-2.0.0+cu117\libtorch\lib这里面的lib都需要引用,且部署时,这里面的DLL都需要拷贝到exe路径下;
https://download.pytorch.org/whl/torch/
https://download.pytorch.org/whl/torchaudio/
页:
[1]