WIN10和Ubuntu系统深度学习环境安装(pytorch框架)Anaconda+CUDA+PyTorch+PyCharm

包括4点:
1、安装Anaconda
2、安装CUDA
3、安装PyTorch
4、安装PyCharm
本次由于选择的PyTorch是1.4版本,支持的是CUDA10.1,所以CUDA安装的版本是10.1。

一、安装Anaconda

1、win10

Anaconda官网 https://www.anaconda.com/distribution/
如下图,选择
在这里插入图片描述根据自己的电脑位数进行选择,下载后安装即可。注意一点
在这里插入图片描述
需要勾选这两个选项。
确认安装成功:打开CMD,输入

conda list

如果出现内容,则代表安装成功。

2、Ubuntu

也打开官网,点击相应版本进行下载,下载后安装即可。
在这里插入图片描述

二、安装CUDA

1、WIN10

CUDA10.1官网
在这里插入图片描述
下载后安装,在选项这里勾选那个方块。测试安装成功:在cmd中输入

nvcc -V

如果有结果,则代表安装成功。

2、Ubuntu

CUDA官网
在这里插入图片描述下载后,按照操作提示按照即可。检查成功安装:
在终端输入

nvcc -V

如果有结果,则代表安装成功。

三、安装PyTorch

1、WIN10

官网下载
在这里插入图片描述以管理员身份打开cmd,输入

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

选择y。
如图显示,则代表安装成功:
在这里插入图片描述
检查安装成功与否:在终端输入

python

然后在python环境下,输入

import torch

如果什么都没出现,则代表安装成功。

2、Ubuntu

打开官网
在这里插入图片描述
在终端输入

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

即可。
检查方式和WIN10大同小异。

四、安装PyCharm

1、WIN10

打开官网
在这里插入图片描述
选择右侧的进行下载安装。
在安装时,在下图这一步,要勾选住相应的选项,如图:
在这里插入图片描述
安装完成后,在新建工程时,添加python路径:
点击File->New Project->Existing interpreter
在这里插入图片描述如果下拉菜单没有选项,则需要点“…”在相应路径中查找。
在这里插入图片描述
点即OK即可。
然后进行测试:新建一个“Python File”
在这里插入图片描述

2、Ubuntu

在官网选择Linux进行下载安装。
最后,完成了这些,恭喜可以进行深度学习之旅。

发布了61 篇原创文章 · 获赞 10 · 访问量 2925

猜你喜欢

转载自blog.csdn.net/weixin_42042056/article/details/105235141