- 安装N卡驱动
- CUDA:CUDA Toolkit Downloads | NVIDIA Developer
- 查看CUDA版本:
nvidia-smi
- Pytorch:Start Locally | PyTorch
- Miniconda:Miniconda — Anaconda documentation
操作步骤
1. 更换conda源
参考:anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
2. 使用conda创建Pytorch环境
conda create -n Pytorch python=3.9
3. 安装Pytorch
GUP:CUDA 12.1
conda activate Pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
4. 验证Pytorch是否安装成功
python
import torch
torch.cuda.is_available()
# 响应true表示安装成功
Windows下如何在CMD或者Powershell中使用pip 更新全部的包
-
打开 CMD 或 PowerShell(以管理员身份运行)
-
运行以下命令来更新 pip 到最新版本:
python -m pip install --upgrade pip
-
更新所有的 Python 包:
PowerShell 默认情况下禁止运行脚本,可先运行以下命令来解除限制:Set-ExecutionPolicy RemoteSigned
pip list --outdated | ForEach-Object { pip install --upgrade $_.split()[0] }
linux下如何在命令行中使用pip 更新全部的包
-
确保 pip 是最新版本:
pip install --upgrade pip
-
更新所有包:
使用以下命令更新所有的 Python 包:pip list --outdated | awk '{print $1}' | xargs -n1 pip install --upgrade
- pip list --outdated:列出所有过期的包
- awk '{print $1}':提取出包的名称
- xargs -n1 pip install --upgrade:逐个更新每个包