centos 7 下通过 conda 安装 cuda pytorch

发布时间：2023-04-07 15:15:47 所属栏目：教程来源：

导读：先查看自己的linux上显卡型号：

# lspci | grep -i nvidia
04:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1)
04:00.1 Audio device: NVIDIA Corporation GP102 HD

先查看自己的linux上显卡型号：

# lspci | grep -i nvidia
04:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1)
04:00.1 Audio device: NVIDIA Corporation GP102 HDMI Audio Controller (rev a1)

查看是否有程序占用（如果存在占用，请停掉该程序）

# lsof | grep nvidia
nvidia-mo 443 root cwd DIR 253,0 254 64 /
nvidia-mo 443 root rtd DIR 253,0 254 64 /
nvidia-mo 443 root txt unkNown /proc/443/exe

当然显卡驱动也可以这样安装：（推荐）
sudo yum install nvidia-detect # 安装检查显卡的程序

nvidia-detect -v # 查看需要的显卡驱动型号

Probing for supported NVIDIA devices...
[10de:1b06] NVIDIA Corporation GP102 [GeForce GTX 1080 Ti]
This device requires the current 440.64 NVIDIA driver kmod-nvidia

yum -y install kmod-nvidia # 安装驱动

错误：nvidia-x11-drv-390xx conflicts with nvidia-x11-drv-460.39-1.el7_9.elrepo.x86_64
错误：nvidia-x11-drv-390xx conflicts with nvidia-x11-drv-libs-460.39-1.el7_9.elrepo.x86_64
错误：nvidia-x11-drv conflicts with nvidia-x11-drv-390xx-390.138-1.el7_8.elrepo.x86_64
您可以尝试添加 --skip-broken 选项来解决该问题
** 发现 2 个已存在的 RPM 数据库问题， 'yum check' 输出如下：
dnf-4.0.9.2-1.el7_6.noarch 有缺少的需求 python2-dnf = ('0', '4.0.9.2', '1.el7_6')
orca-3.6.3-4.el7.x86_64 有缺少的需求 pyatspi

卸载冲突的包

yum remove -y nvidia-x11-drv-390xx-390.138-1.el7_8.elrepo.x86_64
yum remove -y nvidia-x11-drv-460.39-1.el7_9.elrepo.x86_64

卸载驱动：
sudo yum remove kmod-nvidia

# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

# nvidia-smi

Failed to initialize NVML: Driver/library version mismatch

（编辑：汽车网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!