pytorch環境

  1. 安裝anaconda(python3.6環境)

下載 anaconda
例如Anaconda3-5.2.0-Windows-x86_64.exe 安裝時候配置設置環境變量python

  1. 安裝PyCharm(python的ide)
    下載連接 有免費的社區版
  2. 建立PyCharm工程

image.png

選擇工程存放的位置 conda環境的位置和版本
建立以後能夠使用命令
conda env list :看到建立的環境
activate xxx :激活xxx的環境
conda install pytorch torchvision -c pytorch :安裝pytorchwindows

  1. 測試
import torch
print("cuda is available:{}".format(torch.cuda.is_available()))
相關文章
相關標籤/搜索