ubuntu 18.04 + cuda-9.1 + cudnn + pytorch 环境搭建

最新推荐文章于 2024-07-20 07:18:28 发布

waxds1110

最新推荐文章于 2024-07-20 07:18:28 发布

阅读量1.6k

点赞数

本文链接：https://blog.csdn.net/waxds1110/article/details/88742418

版权

本文介绍了在Ubuntu 18.04操作系统上，如何为GTX1060 mobile和i5配置CUDA 9.1、cuDNN以及PyTorch的详细步骤。首先，通过安装显卡驱动，确保NVIDIA 1060-Mobile正确识别。接着，安装CUDA 9.1，降级GCC，设置环境变量。随后，安装cuDNN并将其文件复制到对应目录。最后，通过Anaconda部署PyTorch，更新Python到3.6，并使用PyCharm进行开发。

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

1.安装环境

GPU ： GTX1060 mobile

GPU ： i5

OS ： UBUNTU18.04

2.安装过程与注意事项

1.显卡驱动的安装：

安装cuda需要先安装对应的显卡驱动，ubuntu是默认使用一个集成显卡以及开源驱动，打开终端，使用以下指令：

1.$ sudo passwd 密码
2.$ su root     登陆root账户
3.$ ubuntu-drivers devices 显示可用设备（可以看到可使用的nvidia显卡型号）

/
// 输出结果类似下列：
// == /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
// modalias : pci:v000010DEd00001C20sv00001028sd00000825bc03sc00i00
// vendor   : NVIDIA Corporation
// model    : GP106M [GeForce GTX 1060 Mobile]
// driver   : nvidia-driver-390 - distro non-free recommended
// driver