s天利用晚上這點(diǎn)時(shí)間焙畔,進(jìn)行安裝了pytorch蝌焚。
pytorch的安裝有兩種:
1.conda
2.pip
我是用pip 安裝的。
1.登錄
?PyTorch(https://pytorch.org/)
點(diǎn)擊get started
找到適合自己的版本毡鉴,CUDA刁标、平臺(tái)、Python等,然后會(huì)自動(dòng)提示“Run this command” 命令指令進(jìn)行安裝致讥。我的是Python3.6.5(pycharm 2018.3.6 )+CUDA10.0+windows10仅仆,當(dāng)然如果電腦沒有GPU,CUDA就選none垢袱。
我的cuda是重新下載且安裝的墓拜,去cuda官網(wǎng)下載。(所以沒有經(jīng)歷查看NVIDA控制面板请契,我都沒找到)
CUDA官網(wǎng):CUDA Toolkit 10.1 Update 2 Download | NVIDIA Developer
2.按照要求選擇
(1)準(zhǔn)備工作:查看自己電腦的顯卡版本(n卡)
①打開顯卡控制面板
②選擇幫助->系統(tǒng)信息
③選擇組件咳榜,查看顯卡版本,即為9.2.148爽锥,也即CUDA9
(2)按要求選擇
我的是python 3.6+cuda 10+pip+windows
3涌韩、復(fù)制選項(xiàng)下面的兩條命令到cmd安裝
就是Run this Command:
pip3 install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
因國(guó)內(nèi)下載很慢,我加了換源:-i https://pypi.tuna.tsinghua.edu.cn/simple 清華源
安裝成功:
因?yàn)榇颂幬业膒ip 是需要升級(jí)氯夷,所以升級(jí)了一下:
python -m pip install --upgrade pip
輸入命令pip list臣樱,會(huì)看到安裝列表:
4、測(cè)試一下
進(jìn)入python腮考,然后輸入import?torch雇毫,出現(xiàn)“>>>”且沒報(bào)錯(cuò)那就算成功了
import torch導(dǎo)入沒報(bào)錯(cuò)就說(shuō)明安裝成功,torch.cuda.is_available()是True就表示支持GPU啦~
torch.cuda.is_available()是True就表示支持GPU啦~
或者是在pycharm下: