pytorch官网上有conda
和pip
两种下载方式。
用conda
安装时,-c pytorch
需要删除,否则从官网下载,很慢,如
试将其换成豆瓣源,即:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 http://pypi.douban.com/simple/
但是仍然下载不了。
再试着将其换成pip
下载,即
解决方法:
用wheel安装,即pip,这样就可以了
验证:
import torch
print(torch.__version__) # 输出为:1.7.1+cu110
print(torch.cuda.is_available()) # 输出为:True