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

2023-05-28,,

这篇文章主要介绍了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版本对齐的示例分析”这篇文章对大家有帮助,同时也希望大家多多支持本站,关注本站行业资讯频道,更多相关知识等着你来学习!

《pytorch、cuda、python版本对齐的示例分析.doc》

下载本文的Word格式文档,以方便收藏与打印。