pytorch安裝配置(VScode +miniconda+pytorch)


准備工作:

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.新增環境變量

檢查是否安裝成功,pip show 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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM