Windows下安装pytorch-cpu版本后,再安装gpu版本

Windows下安装pytorch-cpu版本后,再安装gpu版本

由于已经安装了cpu版本了,如果再在该环境下安装gpu版本会造成环境污染,因此,再安装gpu版本时,需要再新建一个虚拟环境才能安装成功。

1.首先安装cuda
在安装之前应该先查看一下自己所适配的版本
在这里插入图片描述
打开系统信息,点击组件可以查看到
在这里插入图片描述

然后去官网下载所适配的版本。
这里可以下载到自己电脑所适配的NVIDIA:此处
接下来的步骤网上很多博客都写了,如果自己的网络不佳,记得选择local版本的下载,否则很容易出错哈。此处博主写的就很好
安装完cuda和cudnn后,开始安装pytorch的gpu版本。
2.安装gpu版本
打开anaconda prompt,首先创建虚拟环境:

#创建新环境
conda create --name pytorch-gpu python=3.7
#激活环境
conda activate pytorch-gpu

输入上面两个语句,名字可以随意取。
激活后去pytorch官网链接
在这里插入图片描述

选择适合自己的版本,但是使用conda方式下载,直接用这个语句是不能下载成功的,因为它的服务器在国外,所以要选择下国内的镜像源。
参考该博主

# 添加Anaconda的清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes  

输入上述语句,后输入
conda install pytorch torchvision cudatoolkit=10.2
此处记得把-c pytorch去掉。就可以直接下载了。安装过程中可能cudatoolkit=10.2这个会出现中断,只需要多数入几次conda install pytorch torchvision cudatoolkit=10.2即可!
期间遇到了warning:A newer version of conda exists。如下图所示:
在这里插入图片描述
方法:

conda install conda=4.8.3

这里根据自己最新版本进行修改,可能一次弄会失败,我就失败了一次,第二次就成功了。
然后查看

conda info

进入pycharm选择interpreter
1.选择file-settings
在这里插入图片描述

2.选择project-interpreter在这里插入图片描述

3.选择下图中右边的设置,点ADD
在这里插入图片描述
剩下的可以参考该博主

  • 8
    点赞
  • 50
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
### 回答1: 1. 首先,你需要下载并安装Python。建议下载Python 3.6或以上版本。 2. 接着,你需要安装pip。pip是Python的包管理器,可以方便地安装和管理Python包。 3. 在安装pip之后,你可以使用pip安装pytorch。可以在pytorch官网上找到适合你的操作系统和Python版本安装命令。 4. 安装完成后,你可以在Python中导入pytorch并开始使用它。 ### 回答2: 要在Windows安装PyTorchCPU版,可以按照以下步骤进行操作: 1. 确保已安装Python:首先,确保在Windows上已经安装了合适的Python版本。可以从Python官方网站(https://www.python.org/downloads/windows/)上下载并安装最新版本。 2. 创建虚拟环境(可选):为了隔离不同项目的依赖,可以使用虚拟环境。在命令提示符下,使用以下命令创建并激活虚拟环境: ``` python -m venv myenv myenv\Scripts\activate ``` 3. 安装PyTorch:打开命令提示符,输入以下命令来安装PyTorch: ``` pip install torch ``` 这将从PyTorch的官方源安装最新版本CPU版。 4. 验证安装安装完成后,输入以下命令以验证PyTorch是否成功安装: ``` python >>> import torch >>> print(torch.__version__) ``` 如果输出了安装PyTorch版本号,则说明安装成功。 现在,你已经成功在Windows安装PyTorchCPU版,可以开始使用它来进行深度学习和机器学习任务了。如果你想使用GPU版本,请确保你的计算机具有兼容的NVIDIA GPU,并按照PyTorch官方文档中的指导安装相应的CUDA驱动程序和cuDNN库。 ### 回答3: 要在Windows安装PyTorchCPU版,您可以按照以下步骤进行操作: 1. 首先,请确保您的计算机上已安装Python。您可以从Python官方网站(www.python.org)下载并安装最新版本Python。 2. 打开命令提示符或PowerShell,输入以下命令,使用pip(Python包管理工具)安装PyTorch: ``` pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 这会下载并安装PyTorchCPU版本以及其依赖项。 3. 安装完成后,您可以尝试导入PyTorch来验证安装是否成功。在Python交互式环境(如IDLE或Jupyter Notebook)中,输入以下代码并运行: ```python import torch print(torch.__version__) ``` 如果安装成功,将显示PyTorch版本号。 请注意,安装PyTorch可能需要一些时间,具体时间取决于网络速度和计算机性能。如果遇到安装问题,可以尝试使用国内源或下载预编译的whl文件进行安装。 此外,如果您的计算机有GPU,并且您希望安装PyTorchGPU版本,则需要确保您的计算机上有适当的CUDA驱动程序和GPU计算能力。在安装过程中,您需要使用不同的pip命令进行GPU版本安装。 总而言之,按照上述步骤,您应该能够成功在Windows安装PyTorchCPU版。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值