(一开始以为cuda10.1不能安装torch GPU版,抱着试试的心态,结果成功了)
1、首先在anaconda下配置pytorch环境,网上很多关于配置的教程,读者可自行查找;
2、然后下载cuda10.1和cudnn,安装并配置环境变量,这里网上也有很多教程,读者可自行查找;
3、下载torch1.2.0和torchvision0.4.4;
4、在anaconda prompt下进入pytorch,进入torch1.2.0和torchvision0.4.4的下载目录,首先安装torch:
5、然后安装torchvision:
6、检查安装成功: