Halcom 发表于 2023-4-7 14:34:00

pytorch1.13.1

torch-1.13.1+cu117-cp37
torchvision-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)

Halcom 发表于 2023-4-7 14:45:20

【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路径下;





Halcom 发表于 2024-10-30 20:53:07

https://download.pytorch.org/whl/torch/
https://download.pytorch.org/whl/torchaudio/


页: [1]
查看完整版本: pytorch1.13.1