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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值