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

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 
  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
你可以按照以下步骤在Ubuntu 18.04上安装CUDA 10.0: 1. 首先,确保你的系统已经安装了适当的NVIDIA显卡驱动。你可以通过运行以下命令来检查驱动程序的安装情况: ``` nvidia-smi ``` 如果该命令返回有关NVIDIA显卡的信息,则说明驱动程序已经正确安装。 2. 下载CUDA Toolkit 10.0的安装包。你可以从NVIDIA的官方网站上下载安装包,或者通过命令行下载。在命令行中,使用以下命令下载安装包: ``` wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux ``` 3. 安装CUDA Toolkit。运行以下命令来安装CUDA Toolkit 10.0: ``` sudo sh cuda_10.0.130_410.48_linux ``` 4. 在安装过程中,你将被要求接受许可协议并选择安装选项。按照提示进行操作。 5. 安装完成后,需要配置环境变量。打开终端并编辑`~/.bashrc`文件: ``` nano ~/.bashrc ``` 6. 在文件的末尾添加以下行,然后保存并关闭文件: ``` export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 7. 执行以下命令来使环境变量生效: ``` source ~/.bashrc ``` 8. 最后,你可以验证CUDA的安装是否成功。运行以下命令: ``` nvcc --version ``` 如果返回CUDA版本信息,说明安装成功。 这样,你就成功在Ubuntu 18.04上安装了CUDA 10.0。希望对你有所帮助!如果有任何问题,请随时向我提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值