Pytorch
-
安装winpython
https://winpython.github.io/ -
安装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
- ide
idle是自带的
推荐thonny
安装Pytorch+GPU
1.Linxu显卡驱动安装如下
- nvidia-smi查看目前版本
- ubuntu-drivers devices显示所有版本,recommend结尾的是推荐的最新版
- 卸载旧驱动sudo apt purge nvidia*
- 更新软件源sudo apt update && sudo apt upgrade -y
- *安装 sudo apt install nvidia-driver-XXX
Windows驱动安装 - 从pytorch出发 https://pytorch.org/ 找到 Pytorch的稳定版,建议下载whl安装最大的文件torchvision。上面写pip支持conda不支持。得到compute platiform = cuda12.x。
- Windows 搜索cuda toolkit 12.x download address,找到安装包下载页面,选择好os、架构、os版本、安装类型后可以下载安装
- Pytorch离线WHL下载
- 从pytorch出发 https://pytorch.org/ 找到显卡版本CUDA 12.8(例子)对应pip安装指令如“pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128”
- 去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版本。
- 下载后,用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激活环境
安装的包都会在此环境下