(2023最新)win11 cuda12.0的配置pytorch与tensorflow方法

win11 4060软件配置:

请谨慎安装:
**首先说明的是下面三个软件配置的pytorch是支持CPU和GPU的。Tensorflow仅支持CUP,这是由于2.11以及以后的版本,将不再支持Windows上的GPU,低版本的cuda可以,但是我没有试过,如果对tensorflow有要求的的,还是不要按照下面软件配置进行安装。**pycharm与Anaconda安装最新版即可。
我安装的软件版本:
pycharm-community-2023.2.3
cuda_12.0.1_528.33_windows
Anaconda3-2023.09-0-Windows-x86_64
cudnn-windows-x86_64-8.8.0.121_cuda12-archive

我的虚拟环境安装到D盘,没安装到默认位置C盘。

本文基于以上三个软件安装完毕之后的pytorch和tensorflow的安装,具体软件安装可以类比视频:https://www.bilibili.com/video/BV1ov41137Z8/?p=7&spm_id_from=pageDriver&vd_source=2c46fa5ca5d5e1d14f5cc47a02ff517c 软件版本不一样但是步骤是一样的,如果环境变量没有添加,需要添加Anaconda3的环境变量。

pytorch安装
由于官网没有找到cuda12.0对应的pytorch,故使用cuda11.8对应的包。

  1. 首先激活conda
activate
  1. 创建python3.8以及激活此环境
conda create --prefix=D:\Environments\mytorch python=3.8
conda activate D:\Environments\mytorch
  1. 安装torch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
  1. 可能出现的问题

4.1、 下载中断需要可输入

 conda clean -i

4.2、出现这个问题

CondaSSLError: Encountered an SSL error. Most likely a certificate verification issue

输入

conda config --set ssl_verify false

4.3、 出现这个问题:

CondaError: Downloaded bytes did not match Content-Length
url: https://conda.anaconda.org/pytorch/win-64/pytorch-2.1.0-py3.8_cuda11.8_cudnn8_0.tar.bz2
target_path: D:\Environments\mytorch\pkgs\pytorch-2.1.0-py3.8_cuda11.8_cudnn8_0.tar.bz2

可用下面方法解决
延长时长:

conda config --set remote_read_timeout_secs 600.0`

or

conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/main
 
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/r
 
conda config --add channels http://mirrors.aliyun.com/anaconda/pkgs/msys2
 
conda config --set show_channel_urls yes
conda config --remove-key channels

注意:
如果一直下载不成功,删除此虚拟环境,删除环境需要用这一条语句

conda remove --all --prefix "虚拟环境地址"

重新按照上面步骤进行。
**可能需要下载多次才能安装完全,包比较多,没安装完成就继续输入下面两句,清除缓存以及安装,这个要看网速和运气。

 conda clean -i
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

tensorflow安装:
这三条指令,我直接安装没有报错。我没用GPU

conda create --prefix=D:\Environments\mytensorflow python=3.8
conda activate D:\Environments\mytensorflow
pip install tensorflow==2.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

但是需要注意的是,网上有些关于CPU和GPU的测试代码,可能会出现不支持GPU情况
,这是由于2.11以及以后的版本,将不再支持Windows上的GPU。

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值