最近小编在帮别人调代码的时候遇到一个难缠的问题!!就是在RTX3060显卡中跑pytorch的代码,好家伙,第一次弄这么个问题弄这么久。
在配置高端显卡的cuda版本的时候一定要看好手册办事,RTX3060目前支持CUDA11及以上,这几天把cuda10以及一下试了一个遍,头晕的不行。最后换成cuda11,这代码跑的搜搜的哦!!pytorch1.7+cuda11+cudnn–匹配11的哦!!
代码运行的现象:
配置的cuda版本低下的时候会出现代码卡住、数据往gpu中传输阻塞(换句话说就是代码不运行了,卡住了哦 —coding xu