已有cuda11.2情况下Pytorch 环境配置

PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/

官网可用的下载链接

win/ubuntu

使用conda

# CUDA 10.2
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch -c conda-forge

使用wheel

# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 10.2
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

直接安装11.1版本的就可以。pytorch自带cuda包，不需要和你电脑的cuda一致，只需要你驱动能够兼容11.1。