在win10下用Anaconda搭建pytorch-gpu安装一二

最新推荐文章于 2024-06-13 09:34:53 发布

天下行走patrick

最新推荐文章于 2024-06-13 09:34:53 发布

阅读量198

点赞数

文章标签： anaconda gpu pytorch 深度学习

本文链接：https://blog.csdn.net/ghjnnhjj/article/details/109331452

版权

本文详细介绍了如何在Windows 10上使用Anaconda搭建PyTorch-GPU环境，包括Anaconda的安装、清华镜像的添加、创建虚拟环境、CUDA 9.0及cuDNN的安装，以及PyTorch的安装过程。遇到的下载速度慢和安装问题，通过调整下载方式和环境变量设置得以解决，最终成功安装并验证了PyTorch的GPU可用性。

摘要由CSDN通过智能技术生成

1、安装Anaconda3-5.3.1-Windows-x86_64.exe；

2、在Anaconda Prompt中添加清华镜像；
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

3、在Anaconda Prompt中创建pytorch-gpu虚拟环境；
conda create --name pytorch_gpu python=3.7
注：若镜像源报错，删除C:\用户***.condarc文件，再conda create --name pytorch_gpu python=3.7

4、激活pytorch_gpu虚拟环境；
activate pytorch_gpu

5、下载CUDA 安装9.0版本；
a。取消勾选CUDA中的Visual Studio Integration 且安装时间有点长耐心等待
b。安装完成重新启动
c。在系统变量中添加以下变量（与Path同级位置）
（变量名称=变量值）
CUDA_PATH = C:\Program Fil