pytorch的个版本
时间: 2023-10-01 12:05:09 浏览: 126
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
PyTorch目前的主要版本包括:
1. PyTorch 0.1.x:最初的版本,不再维护和更新。
2. PyTorch 0.2.x:引入了Variable和新的模型定义方式,支持CUDA 8.0和CuDNN 6。
3. PyTorch 0.3.x:加入了TensorBoard支持,优化了性能和API,并引入了分布式训练功能。
4. PyTorch 0.4.x:加入了TorchScript和JIT编译器,支持ONNX格式,改进了分布式训练和内存管理。
5. PyTorch 1.0.x:重构了整个框架,引入了Eager Execution模式,加强了C++ API和混合精度训练支持。
6. PyTorch 1.1.x:引入了TensorFlow模型转换和新的分布式训练API。
7. PyTorch 1.2.x:支持更多的操作符和硬件,如AMD ROCm和Android移动设备,加强了性能和稳定性。
8. PyTorch 1.3.x:加入了TorchServe和TorchElastic等新功能,改进了分布式训练和性能。
9. PyTorch 1.4.x:增加了对Windows平台的支持,提高了CUDA 10.1和CUDNN 7.6的兼容性,加强了分布式训练和ONNX模型转换。
10. PyTorch 1.5.x:引入了新的自动混合精度功能,加强了分布式训练和模型优化。
11. PyTorch 1.6.x:加入了新的C++前端和TorchServe模型服务器,改进了分布式训练和JIT编译器。
12. PyTorch 1.7.x:加入了支持CUDA 11.0和CUDNN 8.0的版本,增强了ONNX模型的导入和导出功能,加强了分布式训练和性能。
阅读全文