windows下安裝pytorch(基於anaconda)

這是參照官方教程，其實官方教程寫的還是挺詳細的。由於其中提到使用N卡來做加速，而虛擬機沒有配置顯卡的功能，所以我直接安裝到物理機的windows上。

CUDA

如果是使用的N卡，那麼下載cuda安裝

anaconda

去下載anaconda選擇自己合適的版本，下載安裝

pytorch

install

由於anaconda是沙盒模式，要從windows的啓動項進入anaconda promote，自行下面進行安裝

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

測試pytorch

輸入下面命令進入python的交互模式

python

輸入下面命令

from __future__ import print_function
import torch
x = torch.rand(2, 3)
print(x)

如果pytorch沒有裝好的話，執行到第二行就會報錯，如果能打印出來矩陣說明安裝ok

測試cuda

torch.cuda.is_available()

打印出來true說明可以使用cuda

PyCharm

發現還是需要重新安裝

conda install -y pytorch torchvision cudatoolkit=10.2 -c pytorch