pytorch的安装
首先扫盲常用的pip和这里的conda有什么不一样:pip 和 conda 什么区别?
-
先跟着这个win10 安装 pytorch,安装conda、cuda和cuDNN。
-
然后参照win10离线安装pytorch和torchvision进行离线安装
-
命令行输入
nvidia-smi
查看cuda版本,在官网pytorch的pip选项查看自己需要下载的包
- 在这里下载上图中对应离线包
- 打开命令行,进入下载好的离线包所在的文件夹
- 输入
pip install 离线包名称
进行安装即可 - 在编译器输入以下代码进行测试
import torch
print(torch.__version__)
# 返回pytorch版本
print(torch.cuda.is_available())
# 返回True即安装成功
更新安装torchtext部分
因为用到torch一般会用到torchtext进行文本处理或者用torchvision进行图像处理,这个时候居然还要安装库,所以刚开始简简单单用
pip install torchtext
进行安装,在过程中居然发现这样会安装torchtext的最新版本,torchtext是要和torch进行匹配的,所以它直接把我的torch更新到了最新版本,意味着我就不能调用gpu了。。。
经过查找和多次尝试后,发现了torch和torchtext的版本对应:本文的torch1.8.1对应是torchtext0.9.1,其他版本在网上可以找到对应
版本对应表
mindspore的安装
之前所用的conda创建了名为‘py37’的虚拟环境,该python环境中安装有pytorch的一系列包,可以调用GPU。
那么mindspore也可以通过创建虚拟环境的方式进行安装使用,也比较方便。
这里有参照该博客:MindSpore安装和训练体验(Windows10)
-
在这里发布版本列表,下载自己的配置适合的安装包。(原本打算使用官方建议的Ubuntu18.04下调用GPU的模式,后来发现虚拟机不可以调用物理机的GPU,如果硬要用,那么GPU就直接给虚拟机了,物理机以后就不能用了。所以选择windows下的cpu版本)
-
同样打开命令行输入
conda create -n mindspore python=3.7.5
创建python版本为3.7.5的名为mindspore的虚拟环境(其实python3.7.5环境下载好,就会弹出来默认可以创建mindspore名称的虚拟环境) -
输入
activate mindspore
进入虚拟环境,进入自己下载的离线包的文件位置 -
输入
pip install mindspore-1.2.1-cp37-cp37m-win_amd64.whl
即可完成安装
-
打开vscode(自己的python编译器),选择mindspore的虚拟环境
-
输入
import mindspore as ms
print(ms.__version__)
# 如果成功,上面应该打印 1.2.1