pytorch安装conda/pip
下载地址
离线安装地址
torch1.6
conda install pytorch1.6.0 torchvision0.7.0 cudatoolkit=10.1 -c pytorch (cuda10.1)
pip install torch1.6.0+cu101 torchvision0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
conda install pytorch1.6.0 torchvision0.7.0 cudatoolkit=10.2 -c pytorch (cuda10.2)
pip install torch1.6.0 torchvision0.7.0
torch1.5.1
CUDA 10.1
conda install pytorch1.5.1 torchvision0.6.1 cudatoolkit=10.1 -c pytorch
CUDA 10.2
conda install pytorch1.5.1 torchvision0.6.1 cudatoolkit=10.2 -c pytorch
CUDA 10.2
pip install torch1.5.1 torchvision0.6.1
CUDA 10.1
pip install torch1.5.1+cu101 torchvision0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html