前言 目前官网不能直接支持安装cuda10.2 安装过程 创建环境 创建了一个python版本为3.8的torch_test的环境 conda create -n torch_test python=3.8 下载对应whl 在https://download.pytorch.org/whl/torc
torch-1.1.1+cpu.dist.info+ c p u 表示 C P U 版本 torch-1.7.1+cu101.dist-info + cu101 表示 CUDA 10.1 3. 如果发现是cpu版本的torch,那么需要把当前的torch及torchvision删掉, 将这四个包删掉(注意:你的包名称可能和我的有所不同,这不影响,只要是cpu版本的就将其删掉) 4. 查看CUDA Version NV...