win10下在anaconda3虚拟环境配置指定 cuda版本 下的 pytorch
win10下在anaconda3虚拟环境配置指定 cuda版本 下的 pytorch
步骤1:确定本机安装的 cuda 版本
打开 anaconda prompt, 并输入以下语句,查看本机的 cuda 版本
nvcc -V
如图所示,版本为 cudatoolkit 10.0.130,即 cuda 10.0:
步骤2:确定该 cuda 版本 对应的 pytorch 版本
进入 Pytorch 官网
先前版本:
https://pytorch.org/get-started/previous-versi
原创
2020-06-22 16:14:24 ·
4139 阅读 ·
5 评论