pytorch安裝及基本用法


20180425更新  安裝pytorch0.4.0:

conda uninstall pytorch 
# 如果是CUDA版本的話 
conda uninstall cuda80 cuda90 
# 如果 已安裝的 PyTorch 版本 >= 0.3 
conda install -c defaults vc=14  
# 安裝官方的包 
conda install -c pytorch pytorch

 

------------------------分割線--------------------

1、安裝   conda install -c peterjc123 pytorch

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

后續等官方發布正式版本,安裝正式版本。

1.1CUDA和CUDA使用版本查詢

CUDA:統一計算設備架構技術,是一種由NVIDIA推出的通用並行計算架構,該架構使GPU能夠解決復雜的計算問題。 它包含了CUDA指令集架構(ISA)以及GPU內部的並行計算引擎。

 

在英偉達官網下載相應顯卡版本安裝,即可使用。

 

版本查詢:打開該目錄下 version.py即可查詢使用版本。   D:\Anaconda3\pkgs\pytorch-0.3.0-py36_0.3.0cu90\Lib\site-packages\torch

2、介紹

官方0.3.0文檔: http://pytorch.org/docs/0.3.0/torch.html

torch包包含多維張量的數據結構和這些數學類型的運算操作。此外,它提供了有效的序列化張量和任意類型的多種實用工具,還有其他有用的工具。

它有一個CUDA對等體,使你能夠在計算能力> 3的NVIDIA GPU上運行你的張量計算。

3、用法

import torch

dir(torch)

 

dir(torch.Tensor)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM