pytorch1.5 win10 GPU 入坑记录
环境:win10,pytorch1.5 cuda版本10.2,Anaconda 5.3.1 Python 3.7
一、下载安装Anaconda3(此步骤很顺利)
二、开始菜单–Anaconda3–Anaconda Prompt,打开之后默认进入base虚拟环境,创建虚拟环境pytorch,在该环境下按照官网教程安装成功
三、cuda和cudnn安装也很顺利,按照官网教程即可 (此处参考https://blog.csdn.net/qq_41021557/article/details/90381247)
(入坑注意:我按照网上教程先创建虚拟环境pytorch,随后在虚拟环境下安装官网方法安装pytorch成功。在该虚拟环境下import torch 和import torchvision均成功。
然而,问题来了:在base环境下导入均失败。随后又转到pytorch虚拟环境,Torch可以导入成功,但是其他库如tqdm, matplotlib均导入失败。经过一下午各种查资料折磨,最后一想为啥不直接在base环境安装pytorch呢?于是在base环境下暗转官网教程将pytorch又装了一次,这次再导入库均成功!)