安装pytorch1.9
pytorch玩转深度学习
共 1033字，需浏览 3分钟
·
2021-07-01 12:48
windows+conda+cuda11.1：
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
windows+conda+cuda10.2：
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
windows+pip+cuda11.1：
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
windows+pip+cuda10.2：
pip3 install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
Linux+conda+cuda11.1：
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
Linux+conda+cuda10.2：
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
Linux+pip+cuda11.1：
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
Linux+pip+cuda10.2：
pip3 install torch torchvision torchaudio
评论