首页 默认分类 正文
  • 本文约1262字,阅读需6分钟
  • 57
  • 0

Pytorch

  1. 安装winpython
    https://winpython.github.io/

  2. 安装torch(离线)
    https://download.pytorch.org/whl/torch/

本地安装
pip install torch-2.7.0+cpu-cp312-cp312-win_amd64.whl
+cpu是仅支持cpu,cp312是python版本=3.12, win_amd64是CPU型号

pip install torchvision torchaudio numpy matplotlib

  1. ide
    idle是自带的
    推荐thonny

安装Pytorch+GPU

1.Linxu显卡驱动安装如下

  1. nvidia-smi查看目前版本
  2. ubuntu-drivers devices显示所有版本,recommend结尾的是推荐的最新版
  3. 卸载旧驱动sudo apt purge nvidia*
  4. 更新软件源sudo apt update && sudo apt upgrade -y
  5. *安装 sudo apt install nvidia-driver-XXX
    Windows驱动安装
  6. 从pytorch出发 https://pytorch.org/ 找到 Pytorch的稳定版,建议下载whl安装最大的文件torchvision。上面写pip支持conda不支持。得到compute platiform = cuda12.x。
  7. Windows 搜索cuda toolkit 12.x download address,找到安装包下载页面,选择好os、架构、os版本、安装类型后可以下载安装
    1. Pytorch离线WHL下载
  8. 从pytorch出发 https://pytorch.org/ 找到显卡版本CUDA 12.8(例子)对应pip安装指令如“pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
  9. 去https://download.pytorch.org/whl/torch/ 找下面文件"torch-2.7.0+cu128-cp312-cp312-manylinux_2_28_x86_64.whl"和"torch-2.7.0+cu128-cp312-cp312-win_amd64.whl"分别是linux和win对应的pytorch。cu128对应显卡驱动版本,cp312对应python版本。2.7.0是torch版本。
  10. 下载后,用pip安装。注意的是不要修改下载文件名,对导致报错。

conda activate env_name 关于虚拟环境 python3 自带
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
pip install transformers accelerate

python -m venv venvname
venvname/Scripts/activate激活环境
安装的包都会在此环境下

评论
更换验证码
echo "

 

Welcome 2025-05-21 21:09:18 | 3.142.92.19

";