Windows下pytorch的安裝與入門

2021-08-16 04:25:41 字數 599 閱讀 1602

上班前的最後乙個寒假,接觸了pytorch框架。由於家裡只有windows筆記本,因此在windows環境下安裝。安裝方法見我安裝後碰到了如下問題,記錄下來以便查閱:

conda install -c peterjc123 pytorch cuda80

報錯:unsatisfiableerror: the following specifications were found to be in conflict:

- pytorch

經多方查閱,發現是conda版本問題,更新conda:

conda update -f conda

conda -v

更新至conda 4.3.30版本,可以順利安裝。

接著再安裝torchvision:

pip install torchvision

為入門pytorch,我參照了mnist的官方例子,編寫了工程框架,分配了config、data、model等資料夾方便調參,傳送門:

給了個mstar資料的例子(雷達目標識別經典資料庫),注意到為灰度影象,而用imagefolder讀入預設為三通道(單通道通過複製擴充套件為三通道),因此在solver.py檔案中需要指定取第0個通道輸入。

windows下PyTorch安裝之路記錄

最近兩天被windows下pytorch的安裝給搞得很煩了,不過在今天終於安裝成功了,如下圖所示 下面詳細說下此次安裝的詳細記錄吧。我的電腦環境是windows10 cuda9.0 cudnn7.1。然後再安裝的過程中,出現了很多http之類的錯誤,而根據我安裝其他python庫的經驗的話,意思就是...

windows10安裝pytorch環境記錄

因為電腦裡安裝了多個版本的python,所以這裡直接使用anaconda安裝,便於管理包 一 安裝visual studio 2013 後面要用到vs開發環境 二 安裝cuda 10.1 安裝時選擇自定義,只安裝第乙個cuda選項就好 因為筆記本自帶的英偉達驅動比這個新 visual studio ...

基於Windows平台的PyTorch版本安裝

安裝平台 windows pytorch安裝 pytorch build 選擇的stable 作業系統 選擇的windows 根據自己的伺服器系統選擇 安裝方式 選擇的pip 第一次安裝使用conda方式,沒有安裝成功 語言 python cuda 要帶gpu加速的話,可以選擇。需要nvidia顯示...