技術標籤：pythonpythonpytorch

準備工作：

1.安裝VScode

2.安裝miniconda

3.安裝CUDA9.0

4.安裝cuDNN7.0

參考文章：python安裝配置（miniconda版）（Win10+VScode +miniconda），CUDA9.0+cuDNN7.0安裝配置

-------------------------------------以下是正文----------------------------

一、安裝pytorch

1.開啟miniconda，建立環境

//pytorch為環境名，3.6為python版本號
conda create -n pytorch python=3.6
 

2.安裝pytorch

新增映象源，注意一行一行輸入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/    //這一個是新增加的
 

然後安裝

conda activate pytorch //啟用環境

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 //安裝cuda9.0對應的pytorch版本

3.新增環境變數

檢查是否安裝成功，pipshow torch

4.新建python檔案，輸入程式碼，並執行

import torch
print(torch.__version__)
print(torch.cuda.is_available())

安裝成功。

【備註】如果執行過程報錯，大概就是找不到numpy，DLL load failed: 找不到指定的模組。

解決方法：用pip重新安裝numpy（若numpy下載過程出現錯誤，是網速問題）

pip uninstall numpy //解除安裝numpy
pip3 install numpy==1.14.0 //用pip重新安裝numpy