本步骤直接跳过anaconda和cuda的安装教程。
创建一个环境,注意的是python需要与接下来下载的torch组件所匹配。
为了避免使用conda或pip3指令可能存在的下载缓慢情况,可以在此网站下载与cuda和python版本所对应的torch组件:
download.pytorch.org/whl/torch_stable.html
需要组织的是下载的组件版本需要与本机部署的cuda版本和所创建的python环境所匹配。
所需下载的是torch torchvision torchaudio 3个组件。
下载完毕后使用pip install进行安装。
问题一:whl is not a supported wheel on this platform.
输入以下命令进行查看:
pip debug --verbose
再次强调,下载的torch需要与python版本所匹配。
如上图所示即表明安装成功。
最后使用以下命令确认环境是否部署成功:
import torchtorch.__version__torch.cuda.is_available()torch.__version__
import torchvisiontorchvision.__version__