conda 虚拟环境下安装cuda版本pytorch教程

一 、用conda 建好自己的虚拟环境，打开pytorch官网https://pytorch.org/  先择要安装的版本

选择要安装的版本，环境选择windows，安装工具选择conda，语言选择python，平台选择CUDA11.6（需要英伟达独立显卡GPU来支持，需要自行查询自己的电脑配置）

 

 

二 安装CUDA11.X，可以去官网下载

三、安装CuDNN

去官网下载https://developer.nvidia.com/cudnn ，需要注册，解压后覆盖到CUDA的目录下，国内很慢

 

 

四、下载PYtorch，可以用conda直接装，赋值命令（要注意自己的版本，直接用下面的命令可能会版本不对，去官网自己赋值命令）

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

 

也可以自己下载whl文件后，用pip安装