小白从零开始配置pytorch环境

成为不掉头发的工程师2024-11-04 0:01

一、下载ANACONDA

笔者选择的是Anaconda3-5.3.0-Windows-x86_64.exe版本。全程安装可以手机开热点，会快一点。

笔者这里是11.3，安装cuda的时候要求小于等于11.3版本，最好是刚好等于，没有的话小于也可以。

打开ANACONDA的命令行窗口，会显示在base环境里，新建一个环境名为pytorch（可自定义）。

复制代码

%%pytorch为环境的名字，可以自定义
conda create -n pytorch

在命令行窗口中输入

复制代码

conda create -n pytorch python=3.6

按提示输入y。

复制代码

conda activate pytorch

复制代码

pip list

没安装的话则不会显示。

在该网址download.pytorch.org/whl/torch_stable.html找到torch-1.3.0+cu92-cp36-cp36m-win_amd64.whl，下载到自定义路径。

笔者下载于E:\app\EDGE DOWNLOAD

输入下面代码，切换路径。

复制代码

cd E:\app\EDGE DOWNLOAD
E:

再输入下面代码安装torch与cu。

复制代码

pip install torch-1.3.0+cu92-cp36-cp36m-win_amd64.whl

复制代码

pip list

复制代码

python

复制代码

import torch
torch.cuda.is_available()

选择版本的网址：

参考上面torch的安装过程，用下面指令，改一下文件名即可。

复制代码

pip install torchvision-0.4.1+cu92-cp36-cp36m-win_amd64.whl

最终效果，pip list查看已下载文件。