GTX 2080 Windows 10 安裝 Pytorch 1.1 GPU版本
更新於2019/5/10,請酌情參考自身條件採納,前期準備如下:
- 操作系統:WIN 10
- 硬體:9700K + RTX 2080
- 顯卡驅動:430.39
- 下載:
- Anaconda3-5.3.1-Windows-x86_64(Py3.7)
- CUDA Toolkit 10.0 Archive
- cuDNN v7.5.1 (April 22, 2019), for CUDA 10.0
安裝步驟
- 安裝Anaconda,一路Next,在Advanced Option選項,勾選添加進環境變數,安裝完成;
# 修改Conda源到中科大鏡像
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
- 安裝CUDA10,選擇一個空路徑安裝,選擇自定義安裝,並只勾選CUDA組件;
- 解壓縮你下載的cuDNN v7.5.1;
- 參照官網,將解壓出來的cuDNN對應的bin/cudnn64_7.dll, include/cudnn64_7.dll, lib/x64/cudnn.lib三個文件,複製到對應CUDA目錄下;y
- Alt+R,輸入cmd,回車進入命令行模式,輸入
conda create -n py37_pytorch_gpu pip python=3.7
創建conda環境; - 在命令行下,鍵入
conda activate py37_pytorch_gpu
; - 在PyTorch官網找到並鍵入對應安裝命令,我的是
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
- 安裝完成後,在py37_pytorch_gpu環境下啟動python;
import torch
?
print(torch.cuda.is_avaliable())
# True
# 大功告成!
10 .當然,如果你習慣在Jupyter Notebook下做分析建模,你需要手動將創建的環境與Jupyter綁定(參考方法)
source activate myenv
python -m ipykernel install --user --name myenv --display-name "Python (myenv)"
?
conda install nb_conda_kernels
推薦閱讀: