因爲電腦裏安裝了多個版本的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配置的鏡像源