pytorch环境搭建(离线方式)
前言
网上搜到的环境安装教程很多,离线安装的方式是我认为成功率最高,速度最快的方式。我把我搜到的教程总结出来。
准备
(1)安装python(建议版本3.7-3.8)
(2)安装Anaconda
(3)安装pycharm
以上三步直接从官网下载,c盘够用的话安装的时候无脑下一步就行,不然就自己分配位置。
使用conda创建环境
(1)查看显卡驱动
在命令行中(win+r 输入cmd打开)输入nvidia-smi,查看CUDA版本。

(2)conda创建环境
打开开始菜单,进入anaconda prompt(右键->更多->以管理员身份运行)。

输入命令:conda create -n name python=xx (name:你为这个环境命的名 ,xx:你的python版本)

之后会有一个选择页面,输入y表示yes

成功后如下,输入conda activate + 你建的环境名,激活环境

3.在环境中安装离线资源包
(1)下载3个资源包:torch,torchvision,torchaudio
网址: https://download.pytorch.org/whl/torch_stable.html
进入后页面如下

例如:我的配置是cuda11.6与python3.7,所以选择:

之后继续下载torchvision和torchaudio ,注意版本对应关系


最后得到:

(2)安装资源包
安装前,先进行换源
打开命令行(win + r cmd),输入命令:
pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/

回到之前建的环境下,输入命令:pip install + 资源包的位置

全部安装完成

检查安装情况。进入python,输入import torch,输入print(torch.__version__) (这里前后各有两个下划线)

4.在pycharm中使用已搭建的环境
打开pycharm

新建项目




