ubantu20.04和3070ti显卡安装pytorch环境

本文详细介绍了如何在Ubuntu20.04系统上安装3070ti显卡驱动,选择了nvidia-driver-515版本,并手动安装了CUDA11.6,跳过了自动安装显卡驱动的过程。接着,配置了CUDA环境变量,并提到了安装CUDNN和Anaconda以及PyTorch的参考链接。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

ubantu20.04和3070ti显卡安装pytorch环境

1.安装显卡驱动

sudo apt-get update
ubuntu-drivers devices                    //显示当前可用驱动版本
sudo apt-get install nvidia-driver-515    //选择版本并按照

当前可用驱动版本与对应表安装显卡驱动版本
对应表请添加图片描述
我这里选择显卡驱动版本515,cuda为11.6.主要考虑到pytorch安装有与cuda11.6对应的版本

2.安装cuda
如果下载的是cuda的deb安装文件,肯定是要自动安装nvidia-driver的,
为了避免这个事情,不用deb了,改用run文件安装。

$ sudo sh cuda_11.1.6_455.32.00_linux.run

要等一会,会弹出一个界面,选continue, 然后有个协议,输入"accept“,
之后会弹出如下界面,把第一项的driver去掉,然后选"install"
请添加图片描述

安装好之后可以看到如下提示
请添加图片描述

需要配置路径,
在~/.bashrc的最后加入如下两行

export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新配置

source ~/.bashrc

确认一下
$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.6, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

参考
添加链接描述
3.安装cudnn和安装anaconda和pytorch
参考
添加链接描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值