pytorch怎么加入环境变量_Pytorch学习笔记1：环境安装和简单测试代码

1.环境安装：

操作系统是win10

CPU是4860HQ

GPU是矿卡nvidia P106-100 6GB显存

需要安装如下软件(python 3.8不用装，因为anaconda已经包含python3.7)：安装的软件包

安装的顺序：

anaconda

cuda10.1(我的驱动版本是432，太低的驱动版本无法支持10.1，驱动太低请升级)

pycharm

这三个包都是双击有图形界面的，这里就不详细说了，一路next，需要添加环境变量的选项要勾选一下。

然后安装pytorch和torchvision

我才用离线安装方式，因为用conda在线安装网速很慢

下面是安装方法：

用管理员权限运行CMD

在CMD窗口里面到达pytorch和torchvision的存放目录

输入命令：conda install --offline 压缩包的名称

全部安装完就算完成环境搭建

2.测试代码

打开pycharm

新建一个项目

新建一个.py文件

里面输入如下代码：

import torch

print('Torch Version:',torch.__version__)

print('CUDA GPU check:',torch.cuda.is_available())

if(torch.cuda.is_available()):

print('CUDA GPU num:', torch.cuda.device_count())

n=torch.cuda.device_count()

while n > 0:

print('CUDA GPU name:', torch.cuda.get_device_name(n-1))

n -= 1

print('CUDA GPU index:', torch.cuda.current_device())代码实拍，注意代码缩进格式，不然python会报错

Torch Version: 1.6.0

CUDA GPU check: True

CUDA GPU num: 1

CUDA GPU name: P106-100

CUDA GPU index: 0程序运行输出结果

最后，各位有问题可以留言，我也是Pytorch初学者，虽然不一定能回答各位的问题，但欢迎互相交流。

3.安装包资源下载

链接：https://pan.baidu.com/s/1YSIuNXK4tDqDFAki_F9Ong

提取码：dlut