一、搭建pytorch環(huán)境
conda create -n pytorch python=3.7
二田晚、安裝pytorch
激活新建的pytorch環(huán)境
conda activate pytorch
安裝最新的pytorch
pip3 install torch===1.3.0 torchvision===0.4.1 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#從清華鏡像源下載安裝,經(jīng)檢驗(yàn)十分的快,此處默認(rèn)是使用 cuda10.1
pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#cuda9.2
pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
#cpu版本
驗(yàn)證輸入python 進(jìn)入
import torch
torch.__version__# 得到結(jié)果'1.3.0'
三、配置 Jupyter Notebook
新建的環(huán)境是沒有安裝安裝ipykernel的所以無(wú)法注冊(cè)到Jupyter Notebook中襟沮,所以先要準(zhǔn)備下環(huán)境
conda install ipykernel#安裝ipykernel(還是在pytorch 環(huán)境下)
python -m ipykernel install --user --name pytorch --display-name Pytorch-for-Deeplearning#寫入環(huán)境
下一步就是定制 Jupyter Notebook
conda activate base#切換回基礎(chǔ)環(huán)境
jupyter notebook --generate-config#創(chuàng)建jupyter notebook配置文件,這里會(huì)顯示創(chuàng)建jupyter_notebook_config.py的具體位置昌腰。
打開文件开伏,修改
c.NotebookApp.notebook_dir = '' 默認(rèn)目錄位置
c.NotebookApp.iopub_data_rate_limit = 100000000 這個(gè)改大一些否則有可能報(bào)錯(cuò)
至此 pytorch的開發(fā)環(huán)境安裝完成,可以在開始菜單中打開Jupyter Notebook 在New 菜單中創(chuàng)建文件時(shí)選擇Pytorch-for-Deeplearning 創(chuàng)建pytorch的相關(guān)開發(fā)環(huán)境了遭商,輸入如下語(yǔ)句固灵,驗(yàn)證可得即安裝成功。
配置pycharm環(huán)境即將下圖的內(nèi)核換為envs/pytorch環(huán)境下的python內(nèi)核即可劫流。
如有遇到安裝問題即可給我留言巫玻。