包括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环境下,输入
扫描二维码关注公众号,回复:
11132574 查看本文章
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进行下载安装。
最后,完成了这些,恭喜可以进行深度学习之旅。