常用conda指令以及win10 CUDA9.2安装pytorch

最新推荐文章于 2025-03-25 17:06:59 发布

鱼忘机

最新推荐文章于 2025-03-25 17:06:59 发布

阅读量2.3k

点赞数 1

分类专栏： python

本文链接：https://blog.youkuaiyun.com/github_37764152/article/details/108191928

版权

python 专栏收录该内容

2 篇文章

订阅专栏

本文提供在Windows 10系统中安装CUDA及匹配版本PyTorch的详细步骤，包括使用pip与conda进行安装的方法，以及如何解决镜像源问题，确保与Python和pip版本的兼容性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

常用conda指令

添加清华镜像

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 --set show_channel_urls yes

conda update python 
conda update conda
conda update anaconda

win10 CUDA9.2安装pytorch

在网站可以找到原先版本获取下载指令
注意 wheel需要和python pip版本相匹配。下载成功。

# CUDA 10.2
pip install torch==1.5.1 torchvision==0.6.1

# CUDA 10.1
pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.5.1+cu92 torchvision==0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.5.1+cpu torchvision==0.6.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

尝试了conda下载，但是显示连接不上网址。在这里也贴出来

# CUDA 9.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=9.2 -c pytorch

# CUDA 10.1
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.1 -c pytorch

# CUDA 10.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 -c pytorch

# CPU Only
conda install pytorch==1.5.1 torchvision==0.6.1 cpuonly -c pytorch

也有的朋友是用的直接下载文件之后在当前路径直接pip install
给出具体链接，直接打开就可以下载
http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
适用于python3.6以及pip3.6版本，如果升级到3.7需要将36更改为37

所以搞这么复杂…还不如直接在ubuntu上一条指令直接下载啊啊啊

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

其他平台和版本的下载指令在这个网站可以找到