Pytorch安裝倒騰了一上午終於搞定,記錄一下安裝過程.
1. 首先嘗試官網的安裝方式,但是網速太慢了. 除去cudnn100, torchvision和pytorch三個文件,其余可以直接從清華鏡像源下載.
這三個包直接進入官網,使用多線程(我用idm下載的,最好在掛個代理或者VPN)下載后,離線安裝.離線安裝命令:
conda install --use-local Pkg_name
2. 安裝完成后報錯,拷貝系統lib目錄下相關文件到需求路徑,測試成功.
執行如下代碼,需要根據安裝路徑適當調整:
cp /usr/lib/libstdc++.so.6 /opt/anaconda/envs/torch_36/lib/python3.6/site-packages/torch/lib/
cp /usr/lib/libgomp.so.1 /opt/anaconda/envs/torch_36/lib/python3.6/site-packages/torch/lib/
拷貝完畢后進行測試,如下:
這樣子測試GPU:
torch.cuda.is_available()
更新時間:2019-1-24 上面一個一個安裝太費事了,還是新建一個虛擬環境,直接用
pip install -r requirements.txt
比較方便.其中torch下載太慢了,我拷貝鏈接,用axel工具下載后離線安裝.