这里写自定义目录标题
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检测一下安装是否成功即可。