在Win 10+RTX 2070上安装并使用PyTorch的流程(2020年8月)

为啥用Windows?因为对ubuntu的远程桌面觉得不爽,想用Windows的远程桌面。这种教程很多朋友都总结过了,但是要么太复杂(像这个小姐姐就写得特别详细:https://www.jianshu.com/p/d7851ffea38b,但这么多步骤我实在懒得做啊),要么有些点还是没说清楚,我就自己简单记录一下。

首先应该是更新显卡驱动,我们通过鲁大师,驱动精灵更新的显卡虽说可以用,但是可能版本不够新。在cmd下输入:

nvidia-smi

可以看到驱动版本号和其他信息。我也是这样查看了之后才发现自己的驱动比网上教程中还低,强迫症感觉必须要更新一下。我们在这里:https://www.nvidia.cn/Download/index.aspx 可以找到最适合自己显卡的驱动,大家下载的时候注意选择台式或者notebook。

更新驱动之后,就面临着装CUDA,cuDNN等等,我曾经在ubuntu上遇到不少坑:https://blog.csdn.net/qysh123/article/details/79977891。看底下网友评论和其他网友介绍,推荐用Anaconda(据网友介绍:能用conda装的尽量用conda (CUDA和cuDNN也能用conda装,并且是装在虚拟环境内,不影响原来的系统),简单快捷省事并且基本能保证兼容性),说实话,我没有自己用过,之前也更多地是用virtualenv,想知道pip+virtualenv和Anaconda到底有啥区别啊?(pip, virtualenv, fabric并称为Pythoner的三大神器),刚好用这个机会体会一下。

在这里:https://www.anaconda.com/products/individual,选择Windows个人版安装。

安装过程中Anaconda并不建议将其加入Path,所以通过开始——Anaconda3——Anaconda Prompt,输入:

conda create --name pytorch anaconda

来创建名为pytorch的虚拟环境。其实吧,我不理解这么输入的逻辑(为啥后面还要再跟个Anaconda,多此一举啊),但大家都是这么做的:https://www.lizenghai.com/archives/32903.html,https://www.jianshu.com/p/ee08ca2c1160,

然后进入到上面创建的虚拟环境中:

activate pytorch

然后我们给这个虚拟环境安装PyTorch和其他库,在这里:https://pytorch.org/get-started/locally/ 查找应该输入的conda命令即可,例如我这里输入:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

安装完后再这个虚拟环境下进入python,然后依次输入:

import torch
import torchvision
print(torch.__version__)

如果有正常输出版本信息的话则说明安装成功。

退出这个虚拟环境可以输入:

conda deactivate

不过写到这里明显有个问题啊,如果我想让IDE也定位到这个环境中应该怎么办呢?按照一些网友的介绍,例如:https://www.cnblogs.com/wzz2500/p/11356158.html,Anaconda的虚拟环境会放在其安装目录下,例如:C:\ProgramData\Anaconda3\envs,可是首先,ProgramData这个目录是隐藏目录,另外,我发现我的电脑里这个目录下也是空的啊:C:\ProgramData\Anaconda3\envs,这种隐藏目录的操作真是逼死强迫症的节奏。最后发现Anaconda把这个环境放在了:C:\Users\%我的用户名%\.conda\envs,呵呵,这个还是个隐藏目录。不得不说,这种设置太不贴心和友好了。做到这里我还是想自己装一遍CUDA+cuDNN了。

你可能感兴趣的:(深度学习,Python技巧,科研工具)