CONDA CUDA Pytorch安装

1. 查看GPU版本

nvidia-smi

升级显卡升级驱动

重新输入 cmd nvidia-smi

2. 查看CUDA版本驱动对应的支持的CUDA ToolKit工具包

下载安装CUDA

CUDA Toolkit Archive | NVIDIA Developer

3. Conda 

Download Now | Anaconda

国内anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

环境变量设置

conda --version
conda info

替换源

C:\Users\Administrator\.condarc

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
envs_dirs:
  - D:\CONDA\conda_env
pkgs_dirs:    
  - D:\CONDA\conda_pkgs

 conda info 检查

4. 安装Pytorch

PyTorch

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

 创建conda 虚拟环境

conda create --name deeplearn python=3.10

conda activate deeplearn

 安装pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia