不同的神经网络有不同的配置，有些从前公开的github项目，一些包升级之后的版本缺少相关的函数，老版本又不兼容，只能在虚拟环境中重新安装环境。
首先理顺一下CUDA，CUDNN，cudatoolkit的关系。安装CUDA就是安装cudatoolkit，CUDA中包含cudatoolkit. 然后安装CUDA对应版本的CUDNN。Ubuntu22.04对应的是gcc11.04,而CUDA10.7需要gcc7。gcc7的使用过程如连接的文章所示。
https://blog.csdn.net/qq_43200656/article/details/132889191
然后就是下载和安装CUDA。找到对应版本的CUDA，点进去，然后再进行相应的选择得到最终的版本，下载下来之后按照命令行进行安装和配置。之后进行CUDNN的安装。这两个步骤都可以参考下面这个连接
Ubuntu20.04安装cuda10.1_ubuntu 20.04 cuda 10.1-CSDN博客
然后安装对应的pytorch
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch
可以在网上查找与CUDA对应的pytorch进行下载和安装
本文转载自: https://blog.csdn.net/wt852561630/article/details/137376967
版权归原作者 一直都是菜鸟 所有， 如有侵权，请联系我们删除。
版权归原作者 一直都是菜鸟 所有， 如有侵权，请联系我们删除。