4070 python torch安裝cuda 11.8 , cudnn 118

woff

環境：
4070 RTX
Nvidia 546.17


1. 先到官方網站
https://pytorch.org/get-started/previous-versions/


2. 到NVIDIA網站下載CUDA 11.8
https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local


3. 安裝conda，先下載Anacondahttps://www.anaconda.com/download


4. 安裝 torch ,到終端機下指令

1. conda create -n python38 python=3.8
   
2. conda activate python38

複製代碼

1. conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

複製代碼

1. pip install opencv-contrib-python

複製代碼

任何版本都可參考官網，下指令 https://pytorch.org/get-started/previous-versions/

1. pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu118

複製代碼

檢查是否安裝完成

1. pip list

複製代碼

python torch cuda

測試

1. import torch
   
2. torch.cuda.is_available()

複製代碼

完成




參考文章
https://stackoverflow.com/questi ... serror-winerror-127