pytorch、torch下载与安装

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
安装PyTorchTorchVision,你可以按照以下步骤进行: 1. 打开官方网站,复制以下命令: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` 这个命令会安装基于CUDA 11.3的PyTorch和相关的库。 2. 如果你想安装CUDA 10.2的版本,官网不再提供相应的安装命令。但你可以在这个链接中找到之前版本的安装命令:[https://pytorch.org/get-started/previous-versions/](https://pytorch.org/get-started/previous-versions/)。不过,根据经验,使用这种方法安装的成功率可能较低,可能会失败。所以我建议你使用以下方法进行安装: 3. 首先,从这个链接下载你需要的PyTorchTorchVision版本:[https://download.pytorch.org/whl/torch_stable.html](https://download.pytorch.org/whl/torch_stable.html)。比如,如果你需要CUDA 10.2的版本,你可以选择下载以下文件:`cu102/torch-1.6.0-cp37-cp37m-win_amd64.whl`和`cu102/torchvision-0.7.0-cp37-cp37m-win_amd64.whl`。 4. 下载完成后,进入下载torch文件夹,并执行`install-deps`中的命令。这些命令会安装PyTorchTorchVision的依赖项。 5. 最后,使用以下命令克隆Torch的仓库: ``` git clone https://github.com/torch/distro.git ~/torch --recursive ``` 这将克隆Torch的仓库到你的主目录下的`torch`文件夹中。 通过按照以上步骤,你应该能够成功安装PyTorchTorchVision。请根据你的需求选择适当的版本和方法进行安装。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [PyTorch最新安装教程——torch1.6 + torchvision0.7(2022.04)](https://blog.csdn.net/retainenergy/article/details/124032768)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* *3* [torchpytorch安装](https://blog.csdn.net/dcz1994/article/details/89461181)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值