最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。

解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行

Windows+pycharm安装pytorch

这里我们发现有CUDA的选项,这就需要我们安装cuda

怎么安装cuda?

首先我们要知道我们的独立显卡的型号:

Windows+pycharm安装pytorch

然后我们要知道我们的显卡是否支持cuda:点击这里

Windows+pycharm安装pytorch 

 发现是支持的,算力为6.1:

Windows+pycharm安装pytorch

接下来就要下载cuda了,选择10.1(最好)或者10.0都行:点击这里 

Windows+pycharm安装pytorch

 然后下载Windows版本的:

Windows+pycharm安装pytorch

 下载完成后:

Windows+pycharm安装pytorch

 接下来就是安装:默认路径就好

Windows+pycharm安装pytorch

 选择自定义安装:

Windows+pycharm安装pytorch

 按照这个来打勾:

Windows+pycharm安装pytorch

 然后配置环境变量:

Windows+pycharm安装pytorch

然后观察安装是否成功:

Windows+pycharm安装pytorch

CUDNN的配置和下载:点击这里

Windows+pycharm安装pytorch

 然后就需要加入一下,创建个账户,问卷该填的填了:这里我下载的是10.0你们可以根据下载了10.1的cuda就下载10.1的cudnn

Windows+pycharm安装pytorch

 接下来就是下载pytorch了,直接用下面的指令的话速度极慢,还容易断连,所以以下方法抛弃

Windows+pycharm安装pytorch

 

我们去 https://download.pytorch.org/whl/torch_stable.html下载响应的torch文件和torchvision文件,注意版本号,一个是1.4.0,一个是0.5.0

Windows+pycharm安装pytorch

下载好之后:

Windows+pycharm安装pytorch

切到目录下:

Windows+pycharm安装pytorch

 执行相应命令:

Windows+pycharm安装pytorch

 Windows+pycharm安装pytorch

 最后安装成功后:检验

Windows+pycharm安装pytorch

大功告成!!!!,大家还有什么疑问欢迎交流

 

相关文章: