pytorch、torch下载与安装
这里选取的是window的,其它系统重新选取就行下载地址1 下载地址2
torch- torchvision- python版本对应关系
1.py3.7是指你的python版本是python3.7;
2.cuda100是指cuda的版本是10.0版本。

如果不知道自己的电脑的cuda版本号是多少,可以在cmd输入以下命令查看:

nvcc -V

用Aaconda Prompt或者cmd命令cd到文件下载的目录进行安装(两种安装方式

pip install torch-1.1.0-cp36-cp36m-win_amd64.whl
conda install --use-local pytorch-1.0.1-py3.7_cuda100_cudnn7_1.tar.bz2

默认安装到Aaconda下,如果想安装到虚拟环境里,可以尝试将文件先放到虚拟环境中(因为安装本地文件需要在同一个文件夹下),这样通过激活虚拟环境,这里还需要在安装代码前加上python -m

python -m pip install  本地文件名

上面是离线安装,如果是在线安装,

conda install  包名
python -m pip install 包名

这里用pip安装其实和conda安装一样,而且pip更快一些,如果想让conda、pip快一些,可以采用清华源,下面是设置方式,在激活的环境中运行,具体见链接1 链接2

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 config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes是从channel中安装包时显示channel的url,这样就可以知道包的安装来源了

下面pip方法只是每次安装时临时用,想和上面长久用也需要设置,不同系统可以按照链接设置

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

这里使用的是vs code软件安装的jupyter运行时出现KernelRestarter: restart failed...问题,试了很多方法,发现升级下ipykernel管用,应该是版本问题吧
先确定是否安装pip install ipykerne再按照下面代码更新

pip install --upgrade ipykernel
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐