本文在原有anaconda去独立创建新环境,先上参考链接:PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】_哔哩哔哩_bilibili
链接是配置环境特别详细的教程。
本文为记录学习使用,因为本人总是忘。
下面开始
打开终端,输入:conda create -n pytorch1.0.0
(pytorch1.0.0是你创建的环境名称,叫什么都可以,看你心情,方便好记我就写的版本名字)
回车后如下图:
输入 : y
回车后如下图:
创建成功,开始往环境里面安装torch版本,首先切换环境
输入:conda activate pytorch1.0.0
回车后如下图所示即切换成功:
(此时可以通过 pip list 命令查看环境内的安装包,之后安装成功也可通过此代码进行查看)
进入pytorch官网:PyTorch
点击以前的版本选择自己需要的版本,复制对应的命令。
将改代码输入到终端回车,就开始安装啦。
(如果没有安装清华镜像源,可以去安装后再执行此命令,否则会特别慢。
CUDA是自己的显卡驱动版本,可以通过命令 nvidia-smi 查看)
运行结束如图所示:
总结一下核心代码流程:
1、创建环境:conda create -n pytorch1.0.0
2、切换环境:conda activate pytorch1.0.0
3、进入网址:Previous PyTorch Versions | PyTorch
4、执行命令,查看是否安装成功(返回为True即表示这台计算机的GPU可以为pytorch使用):
python
>>import torch
>>torch.cuda.is_available()
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)