30/40RTX显卡的torch环境安装手册

安装过程请参考这篇文章：http://t.csdnimg.cn/A6vs8http://t.csdnimg.cn/A6vs8

本篇文章仅仅提供相关资源和i信息，繁琐的安装步骤已经在另外一篇详细说明。 

🍉一、CUDA

CUDA11.8直达下载链接https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

 

我的是Win11,所以这么选择，小伙伴们跟着自己的系统来哦。

🍉二、Cudnn

cudnn8.9.x系列直达下载链接https://developer.nvidia.com/rdp/cudnn-archive

这里面的都可以就是必须选择最后为->for CUDA 11.x

点击后👇

点击箭头所指的那个，也就是第一个，便会自动开始下载。

🍉三、安装constd(自动配好环境）

        发现的宝藏库const，下载这一个就会自动配好一个版本不冲突，比较新的一套库。

cpu:

pip install cnstd[ort-cpu] -i https://mirrors.aliyun.com/pypi/simple

gpu:

pip install cnstd[ort-gpu] -i https://mirrors.aliyun.com/pypi/simple

安装内容大致如下：

#
# This file is autogenerated by pip-compile with Python 3.9
# by the following command:
#
#    pip-compile --output-file=requirements.txt requirements.in
#
--index-url https://mirrors.aliyun.com/pypi/simple
--extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple
--extra-index-url https://pypi.org/simple

aiohttp==3.9.1
    # via
    #   datasets
    #   fsspec
aiosignal==1.3.1
    # via aiohttp
albumentations==1.3.1
    # via -r requirements.in
appdirs==1.4.4
    # via wandb
async-timeout==4.0.3
    # via aiohttp
attrs==23.1.0
    # via aiohttp
certifi==2023.11.17
    # via
    #   requests
    #   sentry-sdk
charset-normalizer==3.3.2
    # via requests
click==8.1.7
    # via
    #   -r requirements.in
    #   cnstd
    #   wandb
cnstd==1.2.3.6
    # via -r requirements.in
coloredlogs==15.0.1
    # via onnxruntime
contourpy==1.2.0
    # via matplotlib
cycler==0.12.1
    # via matplotlib
datasets[vision]==2.16.0
    # via -r requirements.in
dill==0.3.7
    # via
    #   datasets
    #   multiprocess
docker-pycreds==0.4.0
    # via wandb
filelock==3.13.1
    # via
    #   datasets
    #   huggingface-hub
    #   torch
flatbuffers==23.5.26
    # via onnxruntime
fonttools==4.47.0
    # via matplotlib
frozenlist==1.4.1
    # via
    #   aiohttp
    #   aiosignal
fsspec[http]==2023.10.0
    # via
    #   datasets
    #   huggingface-hub
    #   pytorch-lightning
    #   torch
gitdb==4.0.11
    # via gitpython
gitpython==3.1.40
    # via wandb
huggingface-hub==0.20.1
    # via
    #   cnstd
    #   datasets
humanfriendly==10.0
    # via coloredlogs
idna==3.6
    # via
    #   requests
    #   yarl
imageio==2.33.1
    # via scikit-image
importlib-resources==6.1.1
    # via matplotlib
jinja2==3.1.2
    # via torch
joblib==1.3.2
    # via scikit-learn
kiwisolver==1.4.5
    # via matplotlib
lazy-loader==0.3
    # via scikit-image
lightning-utilities==0.10.0
    # via
    #   pytorch-lightning
    #   torchmetrics
markupsafe==2.1.3
    # via jinja2
matplotlib==3.8.2
    # via
    #   cnstd
    #   seaborn
mpmath==1.3.0
    # via sympy
multidict==6.0.4
    # via
    #   aiohttp
    #   yarl
multiprocess==0.70.15
    # via datasets
networkx==3.2.1
    # via
    #   scikit-image
    #   torch
numpy==1.26.2
    # via
    #   -r requirements.in
    #   albumentations
    #   cnstd
    #   contourpy
    #   datasets
    #   imageio
    #   matplotlib
    #   onnx
    #   onnxruntime
    #   opencv-python
    #   opencv-python-headless
    #   pandas
    #   pyarrow
    #   pytorch-lightning
    #   qudida
    #   scikit-image
    #   scikit-learn
    #   scipy
    #   seaborn
    #   shapely
    #   tifffile
    #   torchmetrics
    #   torchvision
onnx==1.15.0
    # via
    #   -r requirements.in
    #   cnstd
onnxruntime==1.16.3
    # via -r requirements.in
opencv-python==4.8.1.78
    # via cnstd
opencv-python-headless==4.8.1.78
    # via
    #   albumentations
    #   qudida
packaging==23.2
    # via
    #   datasets
    #   huggingface-hub
    #   lightning-utilities
    #   matplotlib
    #   onnxruntime
    #   pytorch-lightning
    #   scikit-image
    #   torchmetrics
pandas==2.1.4
    # via
    #   cnstd
    #   datasets
    #   seaborn
pillow==10.1.0
    # via
    #   -r requirements.in
    #   cnstd
    #   datasets
    #   imageio
    #   matplotlib
    #   scikit-image
    #   torchvision
polygon3==3.0.9.1
    # via cnstd
protobuf==4.25.1
    # via
    #   onnx
    #   onnxruntime
    #   wandb
psutil==5.9.7
    # via wandb
pyarrow==14.0.2
    # via datasets
pyarrow-hotfix==0.6
    # via datasets
pyclipper==1.3.0.post5
    # via cnstd
pyparsing==3.1.1
    # via matplotlib
python-dateutil==2.8.2
    # via
    #   matplotlib
    #   pandas
pytorch-lightning==2.1.3
    # via
    #   -r requirements.in
    #   cnstd
pytz==2023.3.post1
    # via pandas
pyyaml==6.0.1
    # via
    #   albumentations
    #   cnstd
    #   datasets
    #   huggingface-hub
    #   pytorch-lightning
    #   wandb
qudida==0.0.4
    # via albumentations
requests==2.31.0
    # via
    #   datasets
    #   fsspec
    #   huggingface-hub
    #   torchvision
    #   wandb
scikit-image==0.22.0
    # via albumentations
scikit-learn==1.3.2
    # via qudida
scipy==1.11.4
    # via
    #   albumentations
    #   cnstd
    #   scikit-image
    #   scikit-learn
seaborn==0.13.0
    # via cnstd
sentry-sdk==1.39.1
    # via wandb
setproctitle==1.3.3
    # via wandb
shapely==2.0.2
    # via cnstd
six==1.16.0
    # via
    #   docker-pycreds
    #   python-dateutil
smmap==5.0.1
    # via gitdb
sympy==1.12
    # via
    #   onnxruntime
    #   torch
threadpoolctl==3.2.0
    # via scikit-learn
tifffile==2023.12.9
    # via scikit-image
torch==2.1.2
    # via
    #   -r requirements.in
    #   cnstd
    #   pytorch-lightning
    #   torchmetrics
    #   torchvision
torchmetrics==1.2.1
    # via
    #   -r requirements.in
    #   pytorch-lightning
torchvision==0.16.2
    # via
    #   -r requirements.in
    #   cnstd
tqdm==4.66.1
    # via
    #   -r requirements.in
    #   cnstd
    #   datasets
    #   huggingface-hub
    #   pytorch-lightning
typing-extensions==4.9.0
    # via
    #   huggingface-hub
    #   lightning-utilities
    #   pytorch-lightning
    #   qudida
    #   torch
    #   wandb
tzdata==2023.3
    # via pandas
unidecode==1.3.7
    # via cnstd
urllib3==2.1.0
    # via
    #   requests
    #   sentry-sdk
wandb==0.16.1
    # via -r requirements.in
xxhash==3.4.1
    # via datasets
yarl==1.9.4
    # via aiohttp
zipp==3.17.0
    # via importlib-resources

# The following packages are considered to be unsafe in a requirements file:
# setuptools

🍉四、如果是GPU，安装适合CUDA11.8的pytorch

        如果刚刚装的是gpu，必须要走这一步，不然无法正常调用GPU。 

conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia