基於Anaconda 3-4.2.0+CUDA9.0 ——Pytorch1.0.1配置

Anaconda 3-4.2.0

CUDA9.0

python3.5

 

 

Pytorch官網地址：https://pytorch.org/get-started/locally/

嚴格意義上還是需要通過命令行執行，我此次採用的是Anaconda Prompt命令行窗口進行的。

具體步驟如下：

1.去官網查看對應版本的命令

比如我的是

CUDA9.0+python3.5

 

在官網裏就可以查到，然後自動匹配合適的pytorch版本，這個不必擔心。

ps.理論上最開始是有自助選擇的，但我的瀏覽器可能有問題，所以自己翻下去去找到的

2.在命令行執行

想在cmd裏執行的就找到對應的pip代碼

此次是在Anaconda環境下配置，

然後會提示：

安心等待下載完成就可以了，

下載速度會比較慢，大概和源有很大關係。

測試安裝成功：

命令行打開python

執行

import torch

運行成功即可。