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())

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