Pytorch安装配置教程

Pytorch安装配置教程

1.下载并安装anaconda3

下载地址https://www.anaconda.com/download/。主要是使用其中的python
Anaconda的使用教程可以参考:http://www.jianshu.com/p/2f3be7781451

2.下载并安装 PyCharm

下载地址https://www.jetbrains.com/pycharm/download/#section=windows
选择community的版本

3.配置pycharm

a. 打开pycharm,点开菜单找到“setting”,在左侧找到并选择编译器配置,下拉选项卡,选择“Show All”:

b. 在弹出窗口中选择“+”号按钮:

c. 在弹出窗口中,选择“System Interpreter”, 然后再右侧下拉列表中,选择对应的anaconda中安装的python版本。(anaconda默认安装路径一般为“C:\ProgramData\Anaconda3\python.exe”)

4.安装pytorch

(1)Windows安装方法:
先安装cuda和cudnn(略过,如果不用cuda加速可以不装)
然后安装pytorch,在anaconda prompt下面,分版本运行以下命令:
Cpu版本(无cuda): conda install pytorch-cpu -c pytorch
Cuda8版本: conda install pytorch cuda80 -c pytorch
Cuda9.0版本: conda install pytorch cuda90 -c pytorch
Cuda9.1版本: conda install pytorch cuda91 -c pytorch
(如果因为网络原因失败,也可以使用离线安装的方式安装pytorch:
下载下来离线包后运行以下命令:
conda install numpy mkl cffi
conda install --offline ./path(其中“./path”为离线安装包的存放路径))

安装torchvision,在anaconda prompt下面,运行以下命令:
pip install torchvision
或者 从资源python setup.py install

(2)Ubuntu安装方法:
conda install pytorch torchvision cuda90 –c pytorch

おすすめ

転載: blog.csdn.net/weixin_44845357/article/details/116238422