一、創(chuàng)建conda環(huán)境
conda create -n pytorch python=3.6 anaconda
二、確定cuda版本
nvcc -V
我使用的版本是9.0
三更振、安裝PyTorch
用的是官網(wǎng)的安裝命令
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
成功安裝
四、驗(yàn)證安裝
命令行輸入python
>>import torch
>>improt torchvision
>>torch.cuda.is_available()
不報(bào)錯(cuò),返回True土陪,配置成功
五、可能出現(xiàn)的問(wèn)題
如果第四步返回的是False肴熏,則需要看下是不是驅(qū)動(dòng)和cuda的版本對(duì)應(yīng)不上鬼雀,在命令行輸入nvidia-smi
384對(duì)應(yīng)的就是9.0,387對(duì)應(yīng)的是9.1蛙吏,其他的對(duì)應(yīng)關(guān)系自己可以查詢下源哩,找到正確對(duì)應(yīng)版本后,很大幾率可以配置成功鸦做。