pytorch应当在anaconda虚拟环境里安装，不然会安装到liunx自带的python里

最新推荐文章于 2024-08-06 15:57:46 发布

贪玩的攻城

最新推荐文章于 2024-08-06 15:57:46 发布

阅读量1.3k

点赞数 1

文章标签： python pytorch 深度学习 ubuntu

本文链接：https://blog.csdn.net/qq_45567823/article/details/126652225

版权

本文详述了在Ubuntu上配置深度学习环境的步骤，包括设置独显模式、安装NVIDIA驱动、CUDA、Anaconda、CUDNN以及在虚拟环境中安装PyTorch-GPU。避免将库安装到系统默认Python中，确保环境隔离。

摘要由CSDN通过智能技术生成

关于ubuntu配置深度学习环境踩的坑，为未来的自己提个醒
pytorch 应当在 anaconda虚拟环境里安装，不然会安装到liunx自带的python里。
如何配置如下：
1、先把电脑设置成独显模式，可以按F2进入bios设置，或者进入windows设置
2、安装nvidia对应的驱动，建议手动安装，在官网需要对于的显卡驱动
3、安装cuda，cuda 是基于nvidia上的一个有助于nvidia显卡跑深度学习网络的一个东东
4、再安装anaconda，并新建一个虚拟环境，且激活
5、在此虚拟环境里安装cudnn（一个针对深度学习的库）
6、在此虚拟环境下安装pytorch-gpu
切记，cudnn 和pytorch-gpu最好都在虚拟环境里安装，免得安装到liunx自带的python里了