pytorch的源码在github上,网址如下:
https://github.com/pytorch/pytorchhttps://github.com/pytorch/pytorch
pytorch最新版源码编译cuda版:
安装cuda版本的需要
- NVIDIA CUDA
- NVIDIA cuDNN v8.5 or above
pytorch的源码在github上,网址如下:
https://github.com/pytorch/pytorchhttps://github.com/pytorch/pytorch
安装cuda版本的需要