4090 windows cuda12.1 環境配置

西北逍遥發表於2024-03-09

nvidia-smi

C:\Users\jin>
C:\Users\jin>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Mon_Apr__3_17:36:15_Pacific_Daylight_Time_2023
Cuda compilation tools, release 12.1, V12.1.105
Build cuda_12.1.r12.1/compiler.32688072_0

C:\Users\jin>
C:\Users\jin>
C:\Users\jin>
C:\Users\jin>nvidia-smi
Sat Mar  9 18:56:12 2024
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 536.23                 Driver Version: 536.23       CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| 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 RTX 4090      WDDM  | 00000000:01:00.0  On |                  Off |
|  0%   34C    P8              18W / 450W |    419MiB / 24564MiB |      1%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+

+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      1220    C+G   ..._8wekyb3d8bbwe\Microsoft.Photos.exe    N/A      |
|    0   N/A  N/A      6044    C+G   C:\Windows\explorer.exe                   N/A      |
|    0   N/A  N/A      7824    C+G   ...x64__8wekyb3d8bbwe\WinStore.App.exe    N/A      |
|    0   N/A  N/A      8304    C+G   ...7.0_x64__kzf8qxf38zg5c\SkypeApp.exe    N/A      |
|    0   N/A  N/A      8844    C+G   ...CBS_cw5n1h2txyewy\TextInputHost.exe    N/A      |
|    0   N/A  N/A      8900    C+G   ....Search_cw5n1h2txyewy\SearchApp.exe    N/A      |
|    0   N/A  N/A      9436    C+G   ...siveControlPanel\SystemSettings.exe    N/A      |
|    0   N/A  N/A     10068    C+G   ...crosoft\Edge\Application\msedge.exe    N/A      |
|    0   N/A  N/A     13292    C+G   ...GeForce Experience\NVIDIA Share.exe    N/A      |
+---------------------------------------------------------------------------------------+

C:\Users\jin>
C:\Users\jin>
C:\Users\jin>

(base) E:\>
(base) E:\>conda create -n wind_2024 python==3.8
Channels:
 - defaults
Platform: win-64
Collecting package metadata (repodata.json): done
Solving environment: done

## Package Plan ##

  environment location: E:\Anaconda\install_2024\envs\wind_2024

  added / updated specs:
    - python==3.8


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    ca-certificates-2023.12.12 |       haa95532_0         127 KB
    openssl-1.1.1w             |       h2bbff1b_0         5.5 MB
    pip-23.3.1                 |   py38haa95532_0         2.8 MB
    python-3.8.0               |       hff0d562_2        15.9 MB
    setuptools-68.2.2          |   py38haa95532_0         933 KB
    sqlite-3.41.2              |       h2bbff1b_0         894 KB
    vc-14.2                    |       h21ff451_1           8 KB
    vs2015_runtime-14.27.29016 |       h5e58377_2        1007 KB
    wheel-0.41.2               |   py38haa95532_0         126 KB
    ------------------------------------------------------------
                                           Total:        27.3 MB

The following NEW packages will be INSTALLED:

  ca-certificates    pkgs/main/win-64::ca-certificates-2023.12.12-haa95532_0
  openssl            pkgs/main/win-64::openssl-1.1.1w-h2bbff1b_0
  pip                pkgs/main/win-64::pip-23.3.1-py38haa95532_0
  python             pkgs/main/win-64::python-3.8.0-hff0d562_2
  setuptools         pkgs/main/win-64::setuptools-68.2.2-py38haa95532_0
  sqlite             pkgs/main/win-64::sqlite-3.41.2-h2bbff1b_0
  vc                 pkgs/main/win-64::vc-14.2-h21ff451_1
  vs2015_runtime     pkgs/main/win-64::vs2015_runtime-14.27.29016-h5e58377_2
  wheel              pkgs/main/win-64::wheel-0.41.2-py38haa95532_0


Proceed ([y]/n)?y


Downloading and Extracting Packages:


Preparing transaction: done
Verifying transaction: done
Executing transaction: done
#
# To activate this environment, use
#
# $ conda activate wind_2024
#
# To deactivate an active environment, use
#
# $ conda deactivate



(base) E:\>
(base) E:\>


(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
Looking in indexes: https://download.pytorch.org/whl/cu121
Collecting torch
  Downloading https://download.pytorch.org/whl/cu121/torch-2.2.1%2Bcu121-cp38-cp38-win_amd64.whl (2454.9 MB)
     ---------------------------------------- 2.5/2.5 GB ? eta 0:00:00
Collecting torchvision
  Downloading https://download.pytorch.org/whl/cu121/torchvision-0.17.1%2Bcu121-cp38-cp38-win_amd64.whl (5.7 MB)
     ---------------------------------------- 5.7/5.7 MB 21.2 MB/s eta 0:00:00
Collecting torchaudio
  Downloading https://download.pytorch.org/whl/cu121/torchaudio-2.2.1%2Bcu121-cp38-cp38-win_amd64.whl (4.1 MB)
     ---------------------------------------- 4.1/4.1 MB 9.6 MB/s eta 0:00:00
Collecting filelock (from torch)
  Downloading https://download.pytorch.org/whl/filelock-3.9.0-py3-none-any.whl (9.7 kB)
Collecting typing-extensions>=4.8.0 (from torch)
  Downloading https://download.pytorch.org/whl/typing_extensions-4.8.0-py3-none-any.whl (31 kB)
Collecting sympy (from torch)
  Downloading https://download.pytorch.org/whl/sympy-1.12-py3-none-any.whl (5.7 MB)
     ---------------------------------------- 5.7/5.7 MB 20.4 MB/s eta 0:00:00
Collecting networkx (from torch)
  Downloading https://download.pytorch.org/whl/networkx-3.2.1-py3-none-any.whl (1.6 MB)
     ---------------------------------------- 1.6/1.6 MB 13.1 MB/s eta 0:00:00
Collecting jinja2 (from torch)
  Downloading https://download.pytorch.org/whl/Jinja2-3.1.2-py3-none-any.whl (133 kB)
     ---------------------------------------- 133.1/133.1 kB 2.0 MB/s eta 0:00:00
Collecting fsspec (from torch)
  Downloading https://download.pytorch.org/whl/fsspec-2023.4.0-py3-none-any.whl (153 kB)
     ---------------------------------------- 154.0/154.0 kB 1.8 MB/s eta 0:00:00
Collecting numpy (from torchvision)
  Downloading https://download.pytorch.org/whl/numpy-1.24.1-cp38-cp38-win_amd64.whl (14.9 MB)
     ---------------------------------------- 14.9/14.9 MB 21.1 MB/s eta 0:00:00
Collecting pillow!=8.3.*,>=5.3.0 (from torchvision)
  Downloading https://download.pytorch.org/whl/pillow-10.2.0-cp38-cp38-win_amd64.whl (2.6 MB)
     ---------------------------------------- 2.6/2.6 MB 11.2 MB/s eta 0:00:00
Collecting MarkupSafe>=2.0 (from jinja2->torch)
  Downloading https://download.pytorch.org/whl/MarkupSafe-2.1.3-cp38-cp38-win_amd64.whl (17 kB)
INFO: pip is looking at multiple versions of networkx to determine which version is compatible with other requirements. This could take a while.
Collecting networkx (from torch)
  Downloading https://download.pytorch.org/whl/networkx-3.0-py3-none-any.whl (2.0 MB)
     ---------------------------------------- 2.0/2.0 MB 10.8 MB/s eta 0:00:00
Collecting mpmath>=0.19 (from sympy->torch)
  Downloading https://download.pytorch.org/whl/mpmath-1.3.0-py3-none-any.whl (536 kB)
     ---------------------------------------- 536.2/536.2 kB 4.8 MB/s eta 0:00:00
Installing collected packages: mpmath, typing-extensions, sympy, pillow, numpy, networkx, MarkupSafe, fsspec, filelock, jinja2, torch, torchvision, torchaudio
Successfully installed MarkupSafe-2.1.3 filelock-3.9.0 fsspec-2023.4.0 jinja2-3.1.2 mpmath-1.3.0 networkx-3.0 numpy-1.24.1 pillow-10.2.0 sympy-1.12 torch-2.2.1+cu121 torchaudio-2.2.1+cu121 torchvision-0.17.1+cu121 typing-extensions-4.8.0

(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>
(wind_2024) E:\>

(wind_2024) E:\>
(wind_2024) E:\>python
Python 3.8.0 (default, Nov  6 2019, 16:00:02) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>>
>>>
>>> import torch
>>>
>>>
>>> print(torch.__version__)
2.2.1+cu121
>>>
>>>
>>> exit()

(wind_2024) E:\>
(wind_2024) E:\>

cuda

cuDNN

###########################

相關文章