动手学深度学习第二版之环境配置(RTX2060+pytorch1.8.1+cuda11.1)

本文介绍了在配备RTX2060显卡的电脑上,如何配置基于CUDA11.1的PyTorch1.8.1深度学习环境。内容包括硬件环境介绍、Miniconda安装、显卡驱动安装及验证,以及在Ubuntu18.04双系统中遇到的问题和解决方法。
摘要由CSDN通过智能技术生成

动手学深度学习第二版之环境配置(RTX2060+pytorch1.8.1+cuda11.1)

1、电脑硬件环境

CPU :i7-10750H

GPU :rtx2060 显存6G

双系统 ubuntu18.04 win10

2、安装miniconda

参考:https://zh-v2.d2l.ai/chapter_installation/index.html

3、安装显卡驱动

首先检查是否安装显卡驱动

nvidia-smi

如果没有安装,则安装

ubuntu-drivers devices
(base) sunshine@sunshine-NH5x-7xDCx-DDx:~$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001F15sv00001558sd00008521bc03sc00i00
vendor   : NVIDIA Corporation
driver   : nvidia-driver-460-server - distro non-free recommended
driver   : nvidia-driver-450 - distro non-free
driver   : nvidia-driver-460 - distro non-free
driver   : nvidia-driver-450-server - distro non-free
driver   : nvidia-driver-455 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

== /sys/devices/pci0000:00/0000:00:14.3 ==
modalias : pci:v00008086d000006F0sv00008086sd000002A4bc02sc80i00
vendor   : Intel Corporation
manual_install: True
driver   : backport-iwlwifi-dkms - distro free

driver : nvidia-driver-460-server - distro non-free recommended

sudo apt install nvidia-driver-460

安装完成后更新

sudo apt upgrade

最后进行重启

使用下面命令查看是否安装成功

nvidia-smi

如果是这样的就成功了

(base) sunshine@sunshine-NH5x-7xDCx-DDx:~$ nvidia-smi
Sat Apr 24 16:51:03 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.23.05    Driver Version: 455.23.05    CUDA Versi(base) sunshine@sunshine-NH5x-7xDCx-DDx:~$ nvidia-smi
Sat Apr 24 16:51:03 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.23.05    Driver Version: 455.23.05    CUDA Version: 11.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce RTX 2060    On   | 00000000:01:00.0 Off |                  N/A |
| N/A   41C    P8     7W /  N/A |   1322MiB /  5934MiB |     10%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

纷繁中淡定

你的鼓励是我装逼的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值