pytorch安装cuda虚环境
本电脑cuda安装版本是12
查看命令为:
nvidia-smi
Thu Apr 13 15:18:16 2023
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 527.56 Driver Version: 527.56 CUDA Version: 12.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... WDDM | 00000000:01:00.0 On | N/A |
| 0% 45C P8 14W / 130W | 1224MiB / 8192MiB | 6% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1852 C+G ..._dt26b99r8h8gj\RtkUWP.exe N/A |
| 0 N/A N/A 2772 C+G ...bbwe\Microsoft.Photos.exe N/A |
| 0 N/A N/A 3008 C+G ...8.3-Beta-x64\Snipaste.exe N/A |
| 0 N/A N/A 3296 C+G ...\QQLive\QQLiveService.exe N/A |
| 0 N/A N/A 7828 C+G C:\Windows\explorer.exe N/A |
| 0 N/A N/A 8300 C+G ...me\Application\chrome.exe N/A |
| 0 N/A N/A 8716 C+G ...5n1h2txyewy\SearchApp.exe N/A |
| 0 N/A N/A 8728 C+G ...y\ShellExperienceHost.exe N/A |
| 0 N/A N/A 8980 C+G ...5n1h2txyewy\SearchApp.exe N/A |
| 0 N/A N/A 10228 C+G ...e\PhoneExperienceHost.exe N/A |
| 0 N/A N/A 10500 C+G ...2txyewy\TextInputHost.exe N/A |
| 0 N/A N/A 11776 C+G ...lPanel\SystemSettings.exe N/A |
| 0 N/A N/A 11952 C+G ...Client\SunloginClient.exe N/A |
| 0 N/A N/A 13364 C+G ...5\jbr\bin\jcef_helper.exe N/A |
| 0 N/A N/A 15744 C+G ...ge\Application\msedge.exe N/A |
+-----------------------------------------------------------------------------+
显卡NVIDIA Gforce RTX 3050
python版本3.10.10
安装命令
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
结果