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

动手学深度学习第二版之环境配置(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 |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+