cuda10.0 可以正常安装运行pytorch1.7.1

环境:
ubuntu 18.04
conda 4.5.11
gcc 7.5.0
cuda10.0

正常情况下cuda10.0对应的最高pytorch版本为1.4.0 查看,但在cuda10.0的情况下也可以安装高版本的pytorch,并成功运行如下:
在这里插入图片描述
cuda10.0 可以正常安装运行pytorch1.7.1_第1张图片
安装命令,PyToch和Torchvision版本需要对应,参考:

pip install torch==1.7.1 torchvision==0.8.2 -f https://download.pytorch.org/whl/torch_stable.html -i  https://pypi.tuna.tsinghua.edu.cn/simple/

你可能感兴趣的:(软件安装)