来源:自学PHP网 时间:2020-09-27 14:57 作者:小飞侠 阅读:次
[导读] 详解anaconda离线安装pytorchGPU版...
今天带来详解anaconda离线安装pytorchGPU版教程详解
在网速不好的情况下,如何用离线的方式安装pytorch。这里默认大家已经安装了anaconda了。 安装Nvidia驱动、cuda、cudnn等依赖 首先安装vs社区版,如果已经安装过可以跳过这一步,下载地址 之后安装nvidia驱动,注意自己显卡和驱动的对应关系,下载地址 安装cuda 这里要注意查看驱动和cuda的对应关系,首先查看自己下载的驱动文件名, 可以看到最开始有个数字,这个就是驱动版本,和cuda会有下图类似的关系,由于cuda和驱动会不断更新,后面安装的同学,可以去google一下,《cuda和nvidia驱动的对应关系》,获取最新的信息。 这里我下载的是10.2版本的cuda cudnn的安装 确定pytorch安装所需要的包 首先我们创建一个新的conda 环境,在这个环境中我们安装pytorch-gpu版本。 conda create -n pytorch pip python=3.8 pytorch是我们建立的环境名,3.8是该环境的python版本,都可以自己换。 我们在anaconda命令行中输入 conda install pytorch torchvision cudatoolkit=10.2 -c pytorch 会发现有类似的列表
|
自学PHP网专注网站建设学习,PHP程序学习,平面设计学习,以及操作系统学习
京ICP备14009008号-1@版权所有www.zixuephp.com
网站声明:本站所有视频,教程都由网友上传,站长收集和分享给大家学习使用,如由牵扯版权问题请联系站长邮箱904561283@qq.com