pytorch、cuda、python版本对齐的示例分析

这篇文章主要介绍了pytorch、cuda、python版本对齐的示例分析，具有一定借鉴价值，感兴趣的朋友可以参考下，希望大家阅读完这篇文章之后大有收获，下面让小编带着大家一起了解一下。

以pytorch工具为例：

pytorch版本为1.0.1，自带python版本为3.6.2

服务器上GPU的CUDA_VERSION=9000

注意：由于GPU上的CUDA_VERSION为9000，所以至少要安装cuda版本>=9.0，虽然cuda=7.0~8.0也能跑，但是一开始可能会遇到各种各样的问题，本人cuda版本为10.0，安装cuda的命令为：

conda install cudatoolkit=10.0

注：GPU显卡驱动一般没什么问题，所以尽量不要动cudnn的版本。

感谢你能够认真阅读完这篇文章，希望小编分享的“pytorch、cuda、python版本对齐的示例分析”这篇文章对大家有帮助，同时也希望大家多多支持亿速云，关注亿速云行业资讯频道，更多相关知识等着你来学习!