pytorch 安装 cuda 11.1版本

Note: 适用于有梯子,但是用cuda 11.1的版本玩家

在nvidai的官网下载安装好cuda 和cudnn之后,按照网络教程安装好

到pytorch 的官网找到需要安装cuda 11.1 版本的安装命令, 这里默认使用conda安装

pytorch 安装 cuda 11.1版本_第1张图片

但是直接使用conda安装不了cudatoolkit 11.1版本

 

这里推荐的解决版本就是, 直接到anaconda 官网找到上面对应的包, 分别离线安装好

pytorch 安装 cuda 11.1版本_第2张图片

 

 conda install D:\pytorch-1.8.1-py3.7_cuda11.1_cudnn8_0.tar.bz2

其他所有的包都离线安装好之后, 直接测试OK:

pytorch 安装 cuda 11.1版本_第3张图片

你可能感兴趣的:(杂陈)