英伟达显卡的环境安装
- 安装CUDA
- 安装pytorch GPU版本(需要有英伟达显卡,其他命令官网可查)
- conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
- 注意CUDA、显卡、显卡驱动、pytorch之间的版本兼容问题
- CUDA下载:CUDA Toolkit 11.5 Downloads | NVIDIA Developer
- 显卡驱动下载:NVIDIA 驱动程序下载
- Pytorch和CUDA版本匹配查询:Previous PyTorch Versions | PyTorch
- GPU是否支持CUDA:CUDA GPUs | NVIDIA Developer
pytorch的pip安装命令
CPU版本:
# Linux
pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
# Mac & Windows
pip install torch torchvision torchaudio
GPU版本:(安装包超过3g)
# Linux & Windows
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html