windows10下安装cuda对应版本下的pytorch

windows10下安装cuda对应版本的pytorch

首先从图片中了解cuda和pytorch的对应版本关系

我的电脑cuda版本是10.1,之后会下载pytorch1.6.0。

安装步骤

安装步骤：
1、win+R，运行窗口输入cmd打开控制台，输入以下命令

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

使用pip命令让其直接从清华镜像源下载。然后慢慢等待安装成功即可。
最后再进入python检测一下安装是否成功即可。