安装pytorch-gpu版本
安装pytorch1.7.1(GPU版本)
conda换清华源
1 |
|
conda创建一个python=3.7的虚拟环境,并激活
1 |
|
进入该网址下载离线版本的pytorch(gpu) https://download.pytorch.org/whl/torch_stable.html
还有torchversion
(cu102指的是cuda的版本为10.2的,百度查询自己的cuda版本,然后下载对应版本即可,如果没有对应版本,下载比自己cuda版本稍低一点的即可)
在激活的test虚拟环境中完成下面的操作
输入命令:pip install (whl文件保存路径)\文件名称
我刚才下载的两个文件放到了桌面上,所以我的命令如下(改为自己的存储路径即可)
1 |
|
安装完毕后
输入以下命令检查是否安装正确,如图所示打印为True说明安装正确!
本博客所有文章除特别声明外,转载请注明出处!