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