Ubuntu20.04搭建Pytorch1.8深度学习环境

【摘要】 1. 环境版本    Ubuntu 20.04    Python 3.8 (基于anaconda3)    nvidia-driver 460.73    cuda-version 11.2    pytorch 1.8.12. 安装过程1. 安装nvidia驱动sudo apt install nvidia-driver-460    1安装完后重启系统，shell中输入nvidia-s…

1. 环境版本

Ubuntu 20.04
Python 3.8 (基于anaconda3)
nvidia-driver 460.73
cuda-version 11.2
pytorch 1.8.1

2. 安装过程
1. 安装nvidia驱动

sudo apt install nvidia-driver-460

1

安装完后重启系统，shell中输入nvidia-smi 出现如下页面即安装成功

+—————————————————————————–+
| NVIDIA-SMI 460.73.01 Driver Version: 460.73.01 CUDA Version: 11.2 |
|——————————-+———————-+———————-+
| 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 MX150 Off | 00000000:01:00.0 Off | N/A |
| N/A 52C P0 N/A / N/A | 278MiB / 2002MiB | 7% Default |
| | | N/A |
+——————————-+———————-+———————-+

+—————————————————————————–+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID