torch离线安装地址:download.pytorch.org/whl/torch/ torch audio离线安装地址: download.pytorch.org/whl/torchau… torch vision:download.pytorch.org/whl/torchvi…
torch离线安装博客: blog.csdn.net/weixin_4495…
Python torch官方地址:pytorch.org/
anaconda下载地址:www.anaconda.com/download/su… 环境变量: D:\ProgramData\anaconda3 D:\ProgramData\anaconda3\Scripts D:\ProgramData\anaconda3\Library\bin D:\ProgramData\anaconda3\Library\mingw-w64\bin D:\ProgramData\anaconda3\Library\usr\bin
nvidia-smi 检查需要安装nvidia conda的版本 完整博客地址:blog.csdn.net/L_151560241… conda驱动登录查看:docs.nvidia.com/cuda/cuda-t…
conda下载地址:developer.nvidia.com/cuda-toolki…
环境变量
C:\Program Files\NVIDIA Corporation\Nsight Compute 2023.2.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\libnvvp
检查是否成功:nvcc -V
conda info --envs 查看安装环境和位置 conda config --show 直接修改用户目录下的.condarc文件,C:\Users\username eg: channels:
- mirrors.tuna.tsinghua.edu.cn/anaconda/cl…
- mirrors.tuna.tsinghua.edu.cn/anaconda/pk…
- mirrors.tuna.tsinghua.edu.cn/anaconda/pk…
- defaults show_channel_urls: true envs_dirs:
- D:\Users\username\.conda\envs
- D:\ProgramData\anaconda3\envs
- D:\Users\username\AppData\Local\conda\conda\envs pkgs_dirs:
- D:\ProgramData\anaconda3\pkgs
- D:\Users\username\.conda\pkgs
- D:\Users\username\AppData\Local\conda\conda\pkgs
创建并激活当前环境 conda create -n pytorch python=3.11 conda init conda activate pytorch
分别安装3个AI Python包 pip install torch-2.5.0+cu121-cp311-cp311-win_amd64.whl
pip install torchaudio-2.5.0+cu121-cp311-cp311-win_amd64.whl
pip install torchvision-0.20.0+cu121-cp311-cp311-win_amd64.whl
检验安装是否成功
import torch
print(torch.__version__)
print("gpu available:", torch.cuda.is_available())