1. 程式人生 > >64位Win10安裝Pytorch

64位Win10安裝Pytorch

參考:https://zhuanlan.zhihu.com/p/26871672

加清華源,然後使用conda進行安裝。

Pytorch示例:

9:39:49
浪墨 2018/9/6 9:39:49
# -*- coding: utf-8 -*-
'''
Created on 2018年9月6日

@author: user
'''
import torch
import numpy as np

if __name__ == "__main__":  
    # 建立一個np array
    a = np.array([[1,2], [3,4]])
    b = torch.from_numpy(a)      # 根據np array建立torch 張量
    c = b.numpy()                # 根據張量, 匯出np array
    print (c)