AI pytorch环境搭建

574 阅读1分钟

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:

创建并激活当前环境 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())