windows10安裝pytorch環境記錄

因爲電腦裏安裝了多個版本的python,所以這裏直接使用Anaconda安裝,便於管理包

一、安裝visual studio 2013 :
後面要用到vs開發環境

二、安裝cuda-10.1:
安裝時選擇自定義,只安裝第一個CUDA選項就好(因爲筆記本自帶的英偉達驅動比這個新),Visual Studio Integration不要勾選,前面裝了vs了,已經有集成環境了。
在這裏插入圖片描述

三、安裝Anaconda:
下載安裝包(最好去官網根據電腦配置下載最新的,晚上下載速度會快點):官網下載鏈接
在這裏插入圖片描述
注:安裝時一路next就好,路徑可以選擇默認的或者其它的都可以,到下面這一步時選擇下面那個比較好,避免和其它軟件發生衝突。在這裏插入圖片描述
安裝完成以後記得添加路徑到環境變量,如下圖所示:
(Scripts是爲了找到conda)
在這裏插入圖片描述
環境變量生效以後可以在命令行輸入conda info查看信息:
在這裏插入圖片描述
四、安裝pytorch
先去官網查看對應的pytorch版本:
在這裏插入圖片描述
注:直接安裝下載不動,所以這裏通過清華鏡像來安裝

添加鏡像:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/

安裝:

conda install pytorch torchvision cudatoolkit=10.1

安裝完成以後可以查看版本:
在這裏插入圖片描述

注:記一下關於鏡像的幾個命令:

conda config --show  #顯示鏡像源
conda config --remove-key channels   #刪除conda配置的鏡像源

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章