feather的博客

原创 pytorch_gpu环境搭建

安装前 一定 要查看自己电脑的环境配置，然后查询Tensorflow-gpu、Python、 cuda 、 cuDNN 版本关系，要 一 一对应！本次环境的名称为test，在输入命令的时候，指定到test的上一层目录即可，所以为F:\envs\。之后、在激活的虚拟环境中，使用pip安装PyTorch。根据您的需求，选择安装GPU版本。在命令提示窗中进入Python环境，输入。