pytorch(gpu)环境配置+测试环境是否装好

配置环境

1下载cuda和cudnn
参考文章https://zhuanlan.zhihu.com/p/53652964
但是要注意用pip下载比较好,
pip install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp37-cp37m-win_amd64.whl
(.whl可以自己下载,比较快。https://download.pytorch.org/whl/cu100/torch-1.0.1-cp37-cp37m-win_amd64.whl)
用conda下载非常卡,不知道为什么

2.配置虚拟环境安装pytorch
conda create -n wry-pytorch python=3.7
#删除环境
conda remove -n wry-pytorch --all
#激活环境
activate wry-portch
#退出环境
deactivate

##为什么要装虚拟环境?
虚拟环境是一个虚拟化,从电脑独立开辟出来的环境。通俗来说,虚拟环境就是借助虚拟机docker来把一部分内容独立出来。在这个容器里,我们只安装我们依赖的包,各个容器之间互相隔离,比如,这次我想搭建一个pytorch环境,就只需要安装pytorch的相关包就好了;如果要做scrapy库,就开辟一个独立的scrapy库就好了。

3.安装pytorch
pip install torchvision

测试环境是否装好?

1.在cmd中使用命令 conda list 可以看你安装过的包

2.或者,我们可以在安装了pytorch的虚拟环境下输入:python 打开python,并运行:import torch 载入该模块(划重点:加载pytorch这里使用的是import torch 不是import pytorch)**
在这里插入图片描述
没有报错说明安装成功
3.确认你的gpu能不能用。
输入:
torch.cuda.is_available()
显示true,说明安装成功
如果显示是false,可能是显卡驱动没有装上(亲测),可以自行安装一下,我用的驱动人生。

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值