Pytorch在conda+CUDA10.2环境安装
前言
由于之前电脑里安装的CUDA是10.2版本的,paddlepaddle什么的也装的是CUDA10.2的版本,不想再去重装CUDA。
如图:官方已经不支持CUDA10.2的版本了。官网地址:PyTorch
解决
安装过去的版本,点击上面的Previous PyTorch Versions,这里直接给个网址:Previous PyTorch Versions | PyTorch
可以看见有许多过去的版本,也有支持cuda10.2的,大家可以按需找自己要的
完整过程
创建conda虚拟环境
conda create -n env_name python==3.7